Python描述 LeetCode 54. 螺旋矩阵

Posted 2022-06-16 亓官劼

Python描述 LeetCode 54. 螺旋矩阵

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、CSDN、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息.

---

本文原创为【亓官劼】（qí guān jié ），请大家支持原创，部分平台一直在恶意盗取博主的文章！！！ 全部文章请关注微信公众号【亓官劼】。

题目

给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

示例 1：

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

示例 2：

输入：matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出：[1,2,3,4,8,12,11,10,9,5,6,7]

提示：

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

Python描述

模拟题，每次走到头，然后换方向，这里注意下不一定下一个方向就一定能走通，可以连续换3个方向，如果3个方向都不行则说明结束了

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        dr = [[0,1],[1,0],[0,-1],[-1,0]]
        n,m = len(matrix),len(matrix[0])
        x = y = d = 0
        res = []
        while matrix[x][y] != 'v':
            while 0 <= x < n and 0 <= y < m and matrix[x][y] != 'v': 
                res.append(matrix[x][y])
                matrix[x][y] = 'v'
                x, y = x + dr[d][0], y+dr[d][1]
            x, y = x - dr[d][0], y-dr[d][1]
            for _ in range(3):
                d = (d+1)%4
                x, y = x + dr[d][0], y+dr[d][1]
                if 0 <= x < n and 0 <= y < m and matrix[x][y] != 'v':
                    break
        return res