leetcode-59-螺旋矩阵 II

一、题目

给定一个正整数 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