目录
一、题目内容
给你一个正整数 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
二、解题思路
和leetcode_54. 螺旋矩阵类似，只需要按照顺序逐次赋值即可。
三、代码
-
class Solution:
-
def generateMatrix(self, n: int):
-
-
up, down, left, right = 0, n - 1, 0, n - 1 # 上下左右
-
martix = [[0 for _ in range(n)] for _ in range(n)]
-
num = 1
-
while True:
-
# →
-
for i in range(left, right + 1):
-
martix[up][i] = num
-
num += 1
-
up += 1
-
if up > down:
-
break
-
# ↓
-
for i in range(up, down + 1):
-
martix[i][right] = num
-
num += 1
-
right -= 1
-
if right < left:
-
break
-
# ←
-
for i in range(right, left - 1, -1):
-
martix[down][i] = num
-
num += 1
-
down -= 1
-
if down < up:
-
break
-
# ↑
-
for i in range(down, up - 1, -1):
-
martix[i][left] = num
-
num += 1
-
left += 1
-
if left > right:
-
break
-
return martix
-
-
-
if __name__ == '__main__':
-
# n = 3
-
n = 1
-
s = Solution()
-
ans = s.generateMatrix(n)
-
print(ans)
文章来源: nickhuang1996.blog.csdn.net，作者：悲恋花丶无心之人，版权归原作者所有，如需转载，请联系作者。
原文链接：nickhuang1996.blog.csdn.net/article/details/114867993