54. 螺旋矩阵

Problem: 54. 螺旋矩阵

文章目录

• 思路
• 解题方法
• 复杂度
• Code

思路

螺旋矩阵，从左到右、从上到下、从右到左、从下到上，四个方向循环遍历矩阵，需要控制上下左右这四个角。每遍历完成一个方向后，收缩该方向。

• 完成从左到右时：上-=1
• 完成从上到下时：右-=1
• 完成从右到左时：下-=1
• 完成从下到上时：左+=1

解题方法

模拟

复杂度

时间复杂度: O ( m ∗ n ) O(m * n) O(m∗n) 矩阵每个元素只遍历一次

空间复杂度: O ( m ∗ n ) O(m * n) O(m∗n) 矩阵元素个数

Code

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        if not matrix: return []

        l, r, t, b = 0, len(matrix[0]) - 1, 0, len(matrix) - 1

        res = []

        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