Leetcode 59. 螺旋矩阵II

Leetcode 59. 螺旋矩阵II

题目说明

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

示例:

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

题目解析

1. 建立元素全为 0 的正方形矩阵，方向数组((0,1),(1,0),(0,-1)(-1,0))
2. while循环，按方向数组的递进方向给矩阵赋值，当长度超过矩阵长度或矩阵长度坐标位的值不为0，方向改变
3. 时间复杂度O(n), 空间复杂度O($n^2$)

Python代码

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        result = [[0] * n for i in range(n)]
        di_tup = ((0, 1), (1, 0), (0, -1), (-1, 0))
        start, s_x, s_y, di = 1, 0, -1, 0
        while start <= n ** 2:
            x, y = di_tup[di]
            while 0 <= s_x + x < n and 0 <= s_y + y < n and not result[s_x + x][s_y + y]:
                s_x, s_y = s_x + x, s_y + y
                result[s_x][s_y] = start
                start += 1
            di = (di + 1) % 4
        return result