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