Python描述 LeetCode 54. 螺旋矩阵

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

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

Python描述 LeetCode 54. 螺旋矩阵_矩阵

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

题目

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

示例 1:

Python描述 LeetCode 54. 螺旋矩阵_公众号_02

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

示例 2:

Python描述 LeetCode 54. 螺旋矩阵_leetcode_03

输入: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