python每日一题——19螺旋矩阵
题目
给你一个 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]
答案
这道题要求我们按照顺时针螺旋顺序返回矩阵中的所有元素。我们可以使用四个变量来定义矩阵的上下左右边界,然后按照右-下-左-上的顺序遍历矩阵,同时更新这四个边界。
具体实现如下:
def spiralOrder(matrix):
"""
:type matrix: List[List[int]]
:rtype: List[int]
"""
# 定义矩阵的上下左右边界
top, bottom = 0, len(matrix) - 1
left, right = 0, len(matrix[0]) - 1
# 定义结果列表
res = []
# 按照右-下-左-上的顺序遍历矩阵
while left <= right and top <= bottom:
# 右
for i in range(left, right + 1):
res.append(matrix[top][i])
top += 1
# 下
for i in range(top, bottom + 1):
res.append(matrix[i][right])
right -= 1
# 左
if top <= bottom:
for i in range(right, left - 1, -1):
res.append(matrix[bottom][i])
bottom -= 1
# 上
if left <= right:
for i in range(bottom, top - 1, -1):
res.append(matrix[i][left])
left += 1
return res
在这个实现中,我们首先定义了矩阵的上下左右边界,然后按照右-下-左-上的顺序遍历矩阵,同时更新这四个边界。在遍历过程中,我们使用一个列表来保存遍历到的元素,最后返回这个列表即可。