给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
思路:
首先,建立n阶的多维0矩阵,注意不能使用[[0]*n]*n,因为这种方法形成的行向量均存在同一地址。
将坐标设为x,y,当循环次数为圈数,初始p为1,每次将p存入矩阵中后p+1。当遍历圈数为第一行时,每次y+1,最后一列时x+1,最后一行y-1,第一列时x-1,判断当结束完成一圈时终止,开始第二圈的存入数据。
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
matrix = [[0 for i in range(n)] for i in range(n)]
x,y = 0,0
p = 1
for i in range((n+1)//2):
while True:
matrix[x][y] = p0
p += 1
if p == n**2 +1:
break
if y!=n-1-i and x==i:
y+=1
elif y==n-1-i and x!=n-1-i:
x+=1
elif y!=i and x==n-1-i:
y-=1
elif y==i and x!=i+1:
x-=1
else :
y+=1
break
return matrix