Leetcode 54题:螺旋矩阵(Python)
注:本人文章中内容中来自于各大佬的经验总结,仅供自身复习及学习,并无商业用途,若有时忘记标注作者及出处还请提醒,敬请见谅。
题目要求
给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。
示例1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
我的思路
我采取了一种特别简单的思路,既然是螺旋式,那么每次就顺时针输出当前矩阵的最外边一层,然后再将当前矩阵的最外边一层去除,形成一个新的矩阵再次重复输出,以此类推。
就像图片中这样,一层一层输出。
最后我的leetcode代码如下:
class Solution:
def __init__(self):
self.nums = []
def square(self, m, n, mat):
if m > 0 and n > 0:
if m == 1:
for i in range(n):
self.nums.append(mat[0][i])
elif n == 1:
for i in range(m):
self.nums.append(mat[i][0])
else:
# 添加长方形最上面的一行数
for i in range(n-1):
self.nums.append(mat[0][i])
# 添加长方形右侧边缘的数
for i in range(m-1):
self.nums.append(mat[i][n-1])
# 添加长方形底部的一行数
for i in range(n-1, 0,-1):
self.nums.append(mat[m-1][i])
# 添加长方形左侧边缘的一行数
for i in range(m-1, 0, -1):
self.nums.append(mat[i][0])
# 去除矩阵最外面一层边界,进行新一轮循环
mat = mat[1:m-1]
mat = [x[1:n-1] for x in mat]
self.square(m-2, n-2, mat)
def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
m = len(matrix)
n = len(matrix[0])
self.square(m, n, matrix)
res = self.nums
self.nums = []
return res
我的代码水平还有点生涩,感觉还可以再优化,不过终于通过了啊!
其它大佬的思路
来自leetcode的题解里的用户 焦糖小鹦鹉,对边界依次进行遍历。感觉这两种思路本质上一样,但是大佬的代码明显要简洁好看,向他学习啊!
res = []
if matrix is None: return res
top,bottom,left,right = 0, len(matrix) - 1, 0, len(matrix[0]) - 1
while True:
for i in range(left,right+1): #➡️
res.append(matrix[top][i])
top += 1
if top > bottom: break
for i in range(top,bottom+1): #⬇️
res.append(matrix[i][right])
right -= 1
if right < left: break
for i in range(right,left-1,-1): #⬅️
res.append(matrix[bottom][i])
bottom -= 1
if bottom < top: break
for i in range(bottom,top-1,-1): #⬆️
res.append(matrix[i][left])
left += 1
if left > right: break
return res
作者:FvHk1D0CCe
链接:https://leetcode-cn.com/problems/spiral-matrix/solution/xiao-ying-wu-python-54-dong-tai-bian-jie-xuaq/
来源:力扣(LeetCode)
进步!