【LeetCode】螺旋矩阵

一、题目

给你一个 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