给你一个正整数 n ,生成一个包含 1 到 n² 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。 用python
时间: 2024-10-21 15:18:42 浏览: 42
你可以使用Python来创建一个按顺时针螺旋顺序排列的n x n矩阵。下面是一个示例函数,它会接受一个正整数n,并返回对应的螺旋矩阵:
```python
def spiral_matrix(n):
# 初始化矩阵和边界变量
matrix = [[0] * n for _ in range(n)]
top, bottom, left, right = 0, n - 1, 0, n - 1
while top <= bottom and left <= right:
# 从左到右填充顶部行
for i in range(left, right + 1):
matrix[top][i] = i
top += 1
# 从上到下填充底部列
for i in range(top, bottom + 1):
matrix[i][right] = n - 1 - (top - 1) + i
right -= 1
# 如果还有空间,从右到左填充顶部行
if top <= bottom:
for i in range(right, left - 1, -1):
matrix[bottom][i] = n - 2 - (right - 1) + i
bottom -= 1
# 最后从下到上填充底部列
if left >= 0:
for i in range(bottom, top, -1):
matrix[i][left] = left
left += 1
return matrix
# 示例
n = 3
matrix = spiral_matrix(n)
for row in matrix:
print(row)
阅读全文