其他分享
首页 > 其他分享> > [leetcode] 59. Spiral Matrix II

[leetcode] 59. Spiral Matrix II

作者:互联网

题目

Given a positive integer n, generate an n x n matrix filled with elements from 1 to n2 in spiral order.

Example 1:

img

Input: n = 3
Output: [[1,2,3],[8,9,4],[7,6,5]]

Example 2:

Input: n = 1
Output: [[1]]

Constraints:

思路

初始化矩阵,随后遍历。

代码

python版本:

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        matrix = [[0 for _ in range(n)] for _ in range(n)]
        direction = [(0, 1), (1, 0), (0, -1), (-1, 0)]

        def travel(i, j, cnt, d):
            matrix[i][j] = cnt
            x, y = direction[d]
            if 0 <= i+x < n and 0 <= j+y < n and matrix[i+x][j+y] == 0:
                travel(i+x, j+y, cnt+1, d)
            else:
                d = (d+1) % 4
                x, y = direction[d]
                if 0 <= i+x < n and 0 <= j+y < n and matrix[i+x][j+y] == 0:
                    travel(i+x, j+y, cnt+1, d)
                else:
                    return

        travel(0, 0, 1, 0)
        return matrix

# 

标签:cnt,59,Matrix,int,List,direction,II,Input,matrix
来源: https://www.cnblogs.com/frankming/p/16376092.html