一、题目
给定一个正整数 n,生成一个包含 1 到 n^2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例1
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
二、代码
python
def generateMatrix(n):
ans=[[0]*n for i in range(n)]
i,j,di,dj=0,0,0,1
for _ in range(n*n):
if ans[(i+di) % n][(j+dj)% n]!=0:
di,dj=dj,-di
ans[i][j]=_+1
i+=di
j+=dj
return ans