leetcode刷题(螺旋数组)

螺旋矩阵:给定一个正整数nn,生成一个包含 1 到 n2n^2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。

比如:

1 2 3

8 9 4

7 6 5

具体步骤

这种题目需要明白过程的模拟情况。在整个过程中,画矩阵的过程如下:

  • 步骤1:填充上行,从左往右画
  • 步骤2:填充右行,从上往下画
  • 步骤3:填充下行,从右往左画
  • 步骤4:填充左行,由下往上画

起点终点判定

此外,在整个过程中,我们需要注意每次画的过程中的起点和终点。

alt

alt

上面2个图分别是n=5n=5n=6n=6时所形成的螺旋矩阵,里面的黄色底数字则是需要在画矩阵过程中的起始点/终止点。

我们以n=5n=5举例,此外,矩阵中坐标以00为开始

  • 步骤1:其中数字11和数字55的位置分别为(0,0)(0, 0)(0,4)(0, 4),其横坐标一致,但纵坐标的和为4=n14=n-1,同样的,以数字1717和数字1919为例,其位置为(1,1)(1,1)(1,3)(1,3),其横坐标一致,但纵坐标的和为4=n14=n-1。因此,对于步骤1而言,其开始的横坐标为0,1,,n/20, 1, \cdots, \left \lfloor n/2\right \rfloor,纵坐标为0,1,,n/20,1,\cdots,\left\lfloor n/2\right\rfloor

  • 步骤2:我们以数字55和数字99为例,其位置为(0,4)(0,4)(4,4)(4,4)纵坐标一致,但横坐标的和为4=n14=n-1。同样的,数字1919和数字2121,其横坐标一致,但纵坐标的和为4=n14=n-1。因此,对于步骤2而言,其开始的横坐标为n,n1,,nn/2n,n-1,\cdots,n-\left\lfloor n/2\right\rfloor,而纵坐标为0,1,,n/20,1,\cdots,\left\lfloor n/2\right\rfloor

  • 步骤3:我们以数字2323和数字2121为例,其位置为(3,1)(3,1)(3,3)(3,3),其横坐标一致,但纵坐标的和为4=n14=n-1。(省略了,不写)。因此,对于步骤3而言,其开始的横坐标为n,n1,,nn/2n,n-1,\cdots,n-\left\lfloor n/2\right\rfloor,纵坐标为n,n1,,nn/2n,n-1,\cdots,n-\left\lfloor n/2\right\rfloor

  • 步骤4:我们以数字1313和数字11为例,其位置为(4,0)(4, 0)(0,0)(0,0),其纵坐标一致,但横坐标的和为44。(省略了,不写)。因此,对于步骤4而言,其开始的横坐标为0,1,,n/20, 1, \cdots, \left \lfloor n/2\right \rfloor,纵坐标为n,n1,,nn/2n,n-1,\cdots,n-\left\lfloor n/2\right\rfloor

需要注意的是,由于我们每次画圈,均会在上下左右进行一次,因此我们一共会画n/2\left\lfloor n/2\right\rfloor次。而当nn为奇数时,最中间的数字是没法画上去的,因此我们需要对此进行特定的判断以及填充:

if n % 2 != 0:
    matrix[int(n/2)][int(n/2)] = n * n

大循环起点、终点判断:可以看出,每次经历一次步骤1-步骤4后,步骤1的起点位置都会进行变化,分别是从(0,0),(1,1),,(n/2,n/2)(0,0),(1,1),\cdots,(\left\lfloor n/2\right\rfloor,\left\lfloor n/2\right\rfloor)

闭合过程

在这个过程中,我们需要注意矩阵填充过程中是前开后闭还是前闭后开。不过,我们统一采取前闭后开。比如下面这个过程:

alt

这样子就能保证代码编写过程的一致性。

代码

def generateMatrix(self, n: int) -> List[List[int]]:
    """
    n: the size of matrix
    """
    matrix = [[0 for _ in range(n)] for _ in range(n)]
    big_loop = int(n/2)
    start_x = 0
    start_y = 0
    count = 1
    for _ in range(big_loop):
        # 步骤1
        for i in range(start_x, n-start_x-1):
            matrix[start_y][i] = count
            count += 1
        # 步骤2
        for j in range(start_y, n-start_y-1):
            matrix[j][n-start_x-1] = count
            count += 1
        # 步骤3
        for i in range(n-start_y-1, start_x, -1):
            matrix[n-start_y-1][i] = count
            count += 1
        # 步骤4
        for j in range(n-start_x-1, start_x, -1):
            matrix[j][start_y] = count
            count += 1
        start_x += 1
        start_y += 1
    if n % 2 != 0:
        matrix[big_loop][big_loop] = count
    return matrix
全部评论

相关推荐

老粉都知道小猪猪我很久没更新了,因为秋招非常非常不顺利,emo了三个月了,接下来说一下我的情况吧本人是双非本 专业是完全不着计算机边的非科班,比较有优势的是有两段大厂实习,美团和字节。秋招面了50+场泡池子泡死的:滴滴 快手 去哪儿 小鹏汽车 不知名的一两个小厂其中字节13场 两次3面挂 两次2面挂 一次一面挂其中有2场面试题没写出来,其他的都是全a,但该挂还是挂,第三次三面才面进去字节,秋招加暑期总共面了22次字节,在字节的面评可以出成书了快手面了8场,2次实习的,通过了但没去,一次2面挂 最后一次到录用评估 至今无消息滴滴三面完 没几天挂了 所有技术面找不出2个问题是我回答不上来的,三面还来说我去过字节,应该不会考虑滴滴吧,直接给我干傻了去哪儿一天速通 至今无消息小鹏汽车hr 至今无消息美团2面挂 然后不捞我了,三个志愿全部结束,估计被卡学历了虾皮二面挂 这个是我菜,面试官太牛逼了拼多多二面挂 3道题也全写了 也没问题是回答不出来的 泡一周后挂腾讯面了5次 一次2面挂 三次一面挂,我宣布腾讯是世界上最难进的互联网公司然后还有一些零零散散的中小厂,但是数量比较少,约面大多数都是大厂。整体的战况非常惨烈,面试机会少,就算面过了也需要和各路神仙横向对比,很多次我都是那个被比下去的人,不过这也正常,毕竟谁会放着一个985的硕士不招,反而去招一个双非读化学的小子感觉现在互联网对学历的要求越来越高了,不仅仅要985还要硕士了,双非几乎没啥生存空间了,我感觉未来几年双非想要进大厂开发的难度应该直线上升了,唯一的打法还是从大二刷实习,然后苟个转正,不然要是去秋招大概率是炮灰。而且就我面字节这么多次,已经开始问很多ai的东西了,你一破本科生要是没实习没科研懂什么ai啊,纯纯白给了
不知名牛友_:爸爸
秋招你被哪家公司挂了?
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务