【lc刷题】59 螺旋矩阵 II_Day20(69/300)

69/300

  1. 螺旋矩阵 II
    给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
     
    示例:
     
    输入: 3
    输出:
    [
    [ 1, 2, 3 ],
    [ 8, 9, 4 ],
    [ 7, 6, 5 ]
    ]

思路:把空matrix建好了,然后一步步走,一步步填充值。
当走到"把角儿"的时候,右拐。

如何确定走到"把角儿"? 譬如3*3的矩阵第一行,从[0,0]走到[0,2],继续走[0,3]肯定就报错了,这时候用[0%3,3%3]回到[0,0],发现已经被填充,遂拐弯。

而从[0,0]走到[0,2]的过程中,走到[0,1]取模[0%3,1%3]=[0,1]还没被填充,遂填充,继续走,[0,2]取模[0%3,2%3] = [0,2],还没被填充,遂填充,继续走…

同理,3*3的矩阵的最后一行,从[2,2]走到[2,0],再走就是[2,-1],取模[2%3,-1%3]回到[2,2],发现已经被填充,遂拐弯。这里可以复习一下python 负数取模运算

也就是说,对于3*3的矩阵,3就是边界,取模让其保证在0,1,2三个值之中。

如果本来就不超过边界,则为本体;若超过边界,则回到过去点。

文字说不明白,一会儿画个图。

【lc刷题】59 螺旋矩阵 II_Day20(69/300)
画了个丑图,配合代码食用:

3*3的矩阵,从红色的[0][0]开始看起
【lc刷题】59 螺旋矩阵 II_Day20(69/300)

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        matrix = [[0]*n for _ in range(n)]
        
        i, j = 0, 0
        di, dj = 0, 1 #控制direction
        
        for x in range(1, n**2+1):
            matrix[i][j] = x
            if matrix[(i+di)%n][(j+dj)%n]:
                di, dj = dj, -di
            i += di; j += dj
            
        return matrix