【LeetCode】54. 螺旋矩阵
题目:给你一个 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
<= m, n <= 10
-100 <= matrix[i][j] <= 100
解法:
class Solution:
def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
if not matrix or not matrix[0]:
return []
m, n = len(matrix), len(matrix[0])
ans = []
while m > 0 and n > 0:
# 遍历上边
for i in range(n):
ans.append(matrix[0][i])
matrix = matrix[1:]
m -= 1
if m == 0:
break
# 遍历右边
for i in range(m):
ans.append(matrix[i][n-1])
n -= 1
if n == 0:
break
# 遍历下边
for i in reversed(range(n)):
ans.append(matrix[m-1][i])
m -= 1
if m == 0:
break
# 遍历左边
for i in reversed(range(m)):
ans.append(matrix[i][0])
n -= 1
if n == 0:
break
matrix = matrix[1:]
return ans
这段代码定义了一个 spiralOrder
方法,它接收一个二维列表 matrix
作为输入,并返回按照顺时针螺旋顺序排列的元素列表。代码中使用了四个循环来分别遍历矩阵的上、右、下、左边,并通过条件判断避免在边缘情况下的多余遍历。最终返回的 ans
列表即为螺旋顺序的元素结果。
评论已关闭