LeetCode 54-螺旋矩阵

螺旋矩阵

给你一个 m 行 n 列的矩阵 matrix ，请按照顺时针螺旋顺序 ，返回矩阵中的所有元素。

示例 1：
输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

解题思路：按边打印

思路：定义矩阵的四条边，依次打印上、左、下、右四条边。打印每条边后尝试更新边界，如果更新失败，直接退出。

代码：

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> ans = new ArrayList<>();
        if(matrix.length == 0) return ans;
        //1. 定义上下左右四个边界
        int up = 0; //上边界默认为0
        int down = matrix.length - 1; //下边界
        int left = 0; //左边界
        int right = matrix[0].length - 1; //右边界
       
        while(true) {
            //2. 首先输出上边界，从左向右
            for(int i = left; i <= right; i++) {
                ans.add(matrix[up][i]);
            }
            //尝试更新上边界，如果上边界大于下边界，break
            if(++up > down) break;
            //3. 输出右边界，从上至下
            for(int i = up; i <= down; i++) {
                ans.add(matrix[i][right]);
            }
            //尝试更新右边界，如果右边界小于左边界，break
            if(--right < left) break;
            //4. 输出下边界，从右向左
            for(int i = right; i >= left; i--) {
                ans.add(matrix[down][i]);
            }
            //尝试更新下边界，如果下边界小于上边界，break
            if(--down < up) break;
            //5. 输出左边界，从下至上
            for(int i = down; i >= up; i--) {
                ans.add(matrix[i][left]);
            }
            //尝试更新左边界，如果左边界大于右边界，break
            if(++left > right) break;
        }
    
        return ans;
    }
}

时间复杂度：O(mn)。矩阵中的每个元素都要被访问一次

空间复杂度：O(1)。除了输出数组以外，空间复杂度是常数

来源：力扣（LeetCode）链接