其他分享
首页 > 其他分享> > [LeetCode] 54. Spiral Matrix_Medium tag: array, DFS

[LeetCode] 54. Spiral Matrix_Medium tag: array, DFS

作者:互联网

Given an m x n matrix, return all elements of the matrix in spiral order.

 

Example 1:

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

Example 2:

Input: matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
Output: [1,2,3,4,8,12,11,10,9,5,6,7]

 

Constraints:

Ideas:  T: O(m * n),  S: O( m * n)

1. 建一个helper function, 方向为(right,down,left, up), 先从right开始,不停往走,直到走到头或者已经visited过了,再换方向

2. 返回ans

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        self.matrix = matrix
        self.m = len(matrix)
        self.n = len(matrix[0])
        ans = []
        visited = set()
        self.dirs = [(0, 1), (1, 0), ( 0, -1), (-1, 0)]
        self.helper(0, 0, 0, ans, visited)
        return ans
    
    def helper(self, i, j, dire, ans, visited):
        
        visited.add((i, j))
        ans.append(self.matrix[i][j])
        nr, nc = i + self.dirs[dire][0], j + self.dirs[dire][1]
        if 0 <= nr < self.m and 0 <= nc < self.n and (nr, nc) not in visited:
            self.helper(nr, nc, dire, ans, visited)
        else:
            new_dir = (dire + 1) % 4
            nr, nc = i + self.dirs[new_dir][0], j + self.dirs[new_dir][1]
            if 0 <= nr < self.m and 0 <= nc < self.n and (nr, nc) not in visited:
                self.helper(nr, nc, new_dir, ans, visited)

 

标签:dirs,Medium,Matrix,helper,54,self,ans,visited,matrix
来源: https://www.cnblogs.com/Johnsonxiong/p/15171212.html