leetcode算法学习(14)—— 顺时针打印矩阵

题目描述

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字。

示例 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]

限制： 0 <= matrix.length <= 100 0 <= matrix[i].length <= 100

思路

• 定义四个指针：left，right，top，bottom。
• 从四个方向取出矩阵中的数据：
  • 从左至右：(top,left)…(top,right)
  • 从上至下：(top+1,right)…(bottom,right)
  • 从右至左：(bottom,right-1)…(bottom,left)
  • 从下至上：(bottom-1,left)…(top+1,left)
• 外层取出之后，更新top，bottom，left，right的值，进入下一层继续遍历，直到遍历完所有元素为止。

代码

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        res=[]
        if not matrix or not matrix[0]:
            return res
        # 初始化
        rows,cols=len(matrix),len(matrix[0])
        left,top,right,bottom=0,0,cols-1,rows-1
        
        while(top<=bottom and left<=right):
            #从左至右
            for i in range(left,right+1):
                res.append(matrix[top][i])

            #从上至下,至少有两行
            if(top<bottom):
                for i in range(top+1,bottom+1):
                    res.append(matrix[i][right])

            #从右至左,至少2行2列
            if(top<bottom and right>left):
                for i in range(right-1,left-1,-1):
                    res.append(matrix[bottom][i])

            #从下至上,至少3行2列
            if(right>left and bottom>top+1):
                for i in range(bottom-1,top,-1):
                    res.append(matrix[i][left])

            left,top,right,bottom=left+1,top+1,right-1,bottom-1
        return res

复杂度分析

复杂度分析

时间复杂度：O(mn)，其中 mm 和 nn 分别是输入矩阵的行数和列数。矩阵中的每个元素都要被访问一次。
空间复杂度：O(1)，除了输出数组以外，空间复杂度是常数。