LeetCode题解(python)-59. 螺旋矩阵 II

LeetCode题解(python3)

59. 螺旋矩阵 II

题目描述

给定一个正整数 n，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

解题心得

本题是螺旋矩阵I的增强版，只需要使用同样的“拐弯”规则即可。

需要强烈注意的是，矩阵的初始化，你问我为什么matirx = [[0]*n]*n是错误的，而matrix = [0 for _ in range(n)] for _ in range(n)],仔细看看他们的区别，可把我难受坏了……

解题代码

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        matrix = [[0 for _ in range(n)] for _ in range(n)]
        i, j, di, dj = 0, 0, 0, 1
        if matrix != []:
            for t in range(1,n*n+1):
                matrix[i][j] = t
                if matrix[(i + di) % len(matrix)][(j + dj) % len(matrix[0])] != 0:
                    di, dj = dj, -di
                i += di
                j += dj
                print(i,j,di,dj)
        return matrix