54. 螺旋矩阵

54. 螺旋矩阵

原始题目链接：https://leetcode.cn/problems/spiral-matrix/

给你一个 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]:
        row = len(matrix)
        col = len(matrix[0])
        l, r, u, d = 0, col - 1, 0, row - 1
        # 定义一个结果数组
        res = []
        # 当前元素也是个数和总元素个数
        num = row * col

        while num >= 1:
            #从左到右
            for i in range(l, r + 1):
                if num >= 1:
                    res.append(matrix[u][i])
                    num -= 1
            #上边界缩短
            u += 1

            #从上到下
            for i in range(u, d + 1):
                if num >= 1:
                    res.append(matrix[i][r])
                    num -= 1
            r -= 1
            
            #从右到左
            for i in range(r, l - 1, -1):
                if num >= 1:
                    res.append(matrix[d][i])
                    num -= 1
            d -= 1

            #从下到上
            for i in range(d, u - 1, -1):
                if num >= 1:
                    res.append(matrix[i][l])
                    num -= 1
            l += 1

        return res
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

参考文献：
https://leetcode.cn/problems/spiral-matrix/solution/luo-xuan-ju-zhen-by-leetcode-solution/