题目描述:
给你一个 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
思路:
根据题目示例 matrix = [[1,2,3],[4,5,6],[7,8,9]] 的对应输出 [1,2,3,6,9,8,7,4,5] 可以发现,顺时针打印矩阵的顺序是 “从左向右、从上向下、从右向左、从下向上” 循环。
因此,考虑设定矩阵的 “左、右、上、下” 四个边界,模拟以上矩阵遍历顺序。
算法流程:
空值处理: 当 matrix 为空时,直接返回空列表 [] 即可。
初始化: 矩阵 左、右、上、下 四个边界 l , r , t , b ,用于打印的结果列表 res 。
循环打印: “从左向右、从上向下、从右向左、从下向上” 四个方向循环打印。
根据边界打印,即将元素按顺序添加至列表 res 尾部。
边界向内收缩 1 (代表已被打印)。
判断边界是否相遇(是否打印完毕),若打印完毕则跳出。
返回值: 返回 res 即可。
python:
class Solution:
def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
if not matrix:
return []
l,r,t,b,res=0,len(matrix[0])-1,0,len(matrix)-1,[]
while True:
# 左到右,固定上
for i in range(l,r+1):
res.append(matrix[t][i])
t+=1
if t>b:
break
# 上到下,固定右
for i in range(t,b+1):
res.append(matrix[i][r])
r-=1
if l>r:
break
# 右到左,固定下
for i in range(r,l-1,-1):
res.append(matrix[b][i])
b-=1
if t>b:
break
# 下到上,固定左
for i in range(b,t-1,-1):
res.append(matrix[i][l])
l+=1
if l>r:
break
return res
复杂度分析:
时间复杂度 O(MN) : M,N分别为矩阵行数和列数。
空间复杂度 O(1): 四个边界 l , r , t , b 使用常数大小的额外空间。