[LeetCode] 59. Spiral Matrix II tag: array, DFS

2021-08-22 07:31:17 阅读：193 来源： 互联网

标签：dirs 59 matrix helper self Spiral dire Matrix

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

Example 1:

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

Example 2:

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

Constraints:

1 <= n <= 20

Ideas: 跟[LeetCode] 54. Spiral Matrix_Medium tag: array, DFS 的idea一样，只是将visited去掉，可以直接用返回的ans来标记是否visited过

T: O(m * n) S: O(1)

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        self.n = n
        self.matrix = [[-1] * n for _ in range(n)]
        self.dirs = [(0, 1), (1, 0), ( 0, -1), (-1, 0)]
        self.helper(0, 0, 0, 1)
        return self.matrix
    
    
    
    def helper(self, i, j, dire, num):
        
        self.matrix[i][j] = num
        nr, nc = i + self.dirs[dire][0], j + self.dirs[dire][1]
        if 0 <= nr < self.n and 0 <= nc < self.n and self.matrix[nr][nc] == -1:
            self.helper(nr, nc, dire, num + 1)
        else:
            new_dir = (dire + 1) % 4
            nr, nc = i + self.dirs[new_dir][0], j + self.dirs[new_dir][1]
            if 0 <= nr < self.n and 0 <= nc < self.n and self.matrix[nr][nc] == -1:
                self.helper(nr, nc, new_dir, num + 1)

标签：dirs,59,matrix,helper,self,Spiral,dire,Matrix
来源： https://www.cnblogs.com/Johnsonxiong/p/15171217.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

[LeetCode] 59. Spiral Matrix II tag: array, DFS