54. 螺旋矩阵

模拟

import java.util.ArrayList;
import java.util.List;

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {

        /**
         * 定义上下左右边界,每次循环一圈后都要缩小边界
         */
        ArrayList<Integer> list = new ArrayList<>();
        int m = matrix.length;
        int n = matrix[0].length;

        int top = 0;
        int bottom = m - 1;
        int left = 0;
        int right = n - 1;

        int num = m * n;

        while (num >= 1) {

            /**
             * 第一行
             * 因为该矩阵不是方形,会导致重复读取元素,因此用num >= 1来控制读取次数,否则应该结束循环
             */
            for (int i = left; i <= right && num >= 1; i++) {

                list.add(matrix[top][i]);
                num--;
            }

            /**
             * 缩小边界
             */
            top++;

            for (int i = top; i <= bottom && num >= 1; i++) {

                list.add(matrix[i][right]);
                num--;
            }

            right--;

            for (int i = right; i>= left && num >= 1; i--) {

                list.add(matrix[bottom][i]);
                num--;
            }

            bottom--;

            for (int i = bottom; i >= top && num >= 1; i--) {

                list.add(matrix[i][left]);
                num--;
            }

            left++;
        }

        return list;
    }
}

/**
 * 时间复杂度 O(mn)
 * 空间复杂度 O(1)
 */

优化1——便于理解

import java.util.ArrayList;
import java.util.List;

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {

        /**
         * 定义上下左右边界,每次循环一圈后都要缩小边界
         */
        ArrayList<Integer> list = new ArrayList<>();
        int m = matrix.length;
        int n = matrix[0].length;

        int top = 0;
        int bottom = m - 1;
        int left = 0;
        int right = n - 1;

        while (list.size() < m * n) {

            /**
             * 由于不是正方矩阵,遍历到最后一个格子时会存在边界越界,重复读取的情况,因此一旦越界就终止循环
             */

            if (left > right){
                break;
            }

            for (int i = left; i <= right; i++) {
                list.add(matrix[top][i]);
            }

            /**
             * 缩小边界
             */
            top++;

            if (top > bottom){
                break;
            }
            
            for (int i = top; i <= bottom; i++) {
                list.add(matrix[i][right]);
            }

            right--;

            if (left > right){
                break;
            }

            for (int i = right; i>= left; i--) {
                list.add(matrix[bottom][i]);
            }

            bottom--;

            if (top > bottom){
                break;
            }

            for (int i = bottom; i >= top; i--) {
                list.add(matrix[i][left]);
            }

            left++;
        }

        return list;
    }
}

/**
 * 时间复杂度 O(mn)
 * 空间复杂度 O(1)
 */

https://leetcode-cn.com/problems/spiral-matrix/

posted @ 2022-02-17 16:14  振袖秋枫问红叶  阅读(24)  评论(0)    收藏  举报