59. 螺旋矩阵 II
原始题目链接:https://leetcode.cn/problems/spiral-matrix-ii/
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
解题思路:
模拟顺时针进行填充,定义4个边界。具体实现看代码及注释。
代码实现:
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
# 定义左右上下4个边界,进行模拟填充
l, r, u, d = 0, n - 1, 0, n - 1
# 定义一个结果数组
res = [[0 for _ in range(n)] for _ in range(n)]
# 当前元素也是个数和总元素个数
num, total = 1, n * n
# 开始模拟顺时针填充:从左到右 从上到下 从右到左 从下到上
while num <= total:
#从左到右
for i in range(l, r + 1):
res[u][i] = num
num += 1
#上边界缩短
u += 1
#从上到下
for i in range(u, d + 1):
res[i][r] = num
num += 1
r -= 1
#从右到左
for i in range(r, l - 1, -1):
res[d][i] = num
num += 1
d -= 1
#从下到上
for i in range(d, u - 1, -1):
res[i][l] = num
num += 1
l += 1
return res
参考文献:
https://leetcode.cn/problems/spiral-matrix-ii/solution/spiral-matrix-ii-mo-ni-fa-she-ding-bian-jie-qing-x/