Leetcode 59. Spiral Matrix II 螺旋矩阵 II

Leetcode 59. Spiral Matrix II 螺旋矩阵 II

标签： Leetcode

---

题目地址：https://leetcode-cn.com/problems/spiral-matrix-ii/

题目描述

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

示例:

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

算法思想

和之前的那个螺旋矩阵几乎一样，把遍历变成输入进去即可，并没有什么不同的地方，同样用几个方向，根据res是否为None作为判断即可。详情可参考旋转矩阵这篇文章。

python代码

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        res = [[None]*n for _ in range(n)]

        x,y,nx,ny = 0,0,0,1
        for i in range(n*n):
            res[x][y] = i+1
            if not res[(x+nx)%n][(y+ny)%n] is None:# 说明来过了，换方向
                nx, ny = ny, -nx
            x+=nx
            y+=ny
        return res