LeetCode打卡25——螺旋矩阵II

给定一个正整数 n，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        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