给你一个正整数 n
,生成一个包含 1
到 n2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
示例 输入:n = 3 输出:[[1,2,3],[8,9,4],[7,6,5]]
这道题想了很久都没有思路,最后还是参考了官方答案看了题解才写出来:
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
left, right, top, bottom = 0, n-1, 0, n-1
matric = [[0 for _ in range(n)]for _ in range(n)]
count, target = 1, n*n
while count <= target:
for i in range(left, right + 1): #从左到右
matric[top][i] = count
count += 1
top += 1
for i in range(top, bottom + 1): #从上到下
matric[i][right] = count
count += 1
right -= 1
for i in range(right, left - 1, -1): #从右到左
matric[bottom][i] = count
count += 1
bottom -= 1
for i in range(bottom, top - 1, -1): #从下到上
matric[i][left] = count
count += 1
left += 1
return matric