目录
- 一、题目
- 二、解法
- 完整代码
一、题目
给你一个 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
二、解法
就模拟就好了,顺时针模拟
但是要注意各种变量,什么时候加,什么时候减,最好拿笔画一下,不然容易搞乱
完整代码
class Solution:
def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
m, n = len(matrix), len(matrix[0])
l, r, t, b = 0, n - 1, 0, m - 1
cnt = m * n
res = []
while cnt > 0:
i = l
# 上边
while i <= r and cnt > 0:
res.append(matrix[t][i])
cnt -= 1
i += 1
t += 1
i = t
# 右边
while i <= b and cnt > 0:
res.append(matrix[i][r])
cnt -= 1
i += 1
r -= 1
i = r
# 下边
while i >= l and cnt > 0:
res.append(matrix[b][i])
cnt -= 1
i -= 1
b -= 1
i = b
# 左边
while i >= t and cnt > 0:
res.append(matrix[i][l])
cnt -= 1
i -= 1
l += 1
return res