螺旋矩阵 II（LeetCode）

题目

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

解题

def generateMatrix(n):matrix = [[0] * n for _ in range(n)]top, bottom = 0, n - 1left, right = 0, n - 1num = 1while top <= bottom and left <= right:# 从左到右填充for i in range(left, right + 1):matrix[top][i] = numnum += 1top += 1# 从上到下填充for i in range(top, bottom + 1):matrix[i][right] = numnum += 1right -= 1if top <= bottom:# 从右到左填充for i in range(right, left - 1, -1):matrix[bottom][i] = numnum += 1bottom -= 1if left <= right:# 从下到上填充for i in range(bottom, top - 1, -1):matrix[i][left] = numnum += 1left += 1return matrix# 测试
n = 3
result = generateMatrix(n)
for row in result:print(row)