ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

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

2021-08-22 07:32:07  阅读:204  来源: 互联网

标签:dirs Medium Matrix helper 54 self ans visited matrix


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:

  • m == matrix.length
  • n == matrix[i].length
  • 1 <= m, n <= 10
  • -100 <= matrix[i][j] <= 100

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

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

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有