59. 螺旋矩阵 II
题目
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
1 <= n <= 20
思路
一圈一圈来,与此题螺旋矩阵的方法二一样。
代码
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
res = []
for i in range(n):
t= [0]*n
res.append(t)
print(res)
cout = (n+1)/2
i = 0
number = 1
while(i < cout):
for j in range(i,n-i):
res[i][j]=number
number+=1
for j in range(i+1,n-i):
res[j][n-i-1]= number
number+=1
if(i!=n-1-i):
for j in range(n-i-2,i-1,-1):
res[n-i-1][j]=number
number+=1
if(i !=n-1-i):
for j in range(n-2-i,i,-1):
res[j][i]=number;
number+=1
i+=1
return res