【LeetCode刷题】螺旋矩阵 Java实现

【LeetCode刷题】螺旋矩阵 Java实现

题目：

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

示例1：

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

示例2：

输入:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

我的思路：

• 将螺旋遍历分为两部分：顺时针遍历和逆时针遍历
• 顺时针临界条件就是行和列都没有超过最大值-N 逆时针临界条件就是行不小于N+1，列不小于N
• N初始值为0，每完成一圈的顺时针和逆时针遍历就将N加1
• 通过标志位来掉转遍历方向，注意每次改变方向时都会导致程序做了一次无效的遍历，因此要补回遍历次数

代码：

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        if(matrix == null || matrix.length == 0){
            return new ArrayList<>();
        }
        int rows = matrix.length;
        int cols = matrix[0].length;
        int total = rows * cols;
        List<Integer> numList = new ArrayList<>();
        int i = 0, j = 0, n = 0;
        boolean flag = true;
        numList.add(matrix[0][0]);
        for (int k = 0; k < total; k++) {
            if (flag) {
                if (j < cols - n - 1) {
                    j++;
                    numList.add(matrix[i][j]);
                    continue;
                }
                if (i < rows - n - 1) {
                    i++;
                    numList.add(matrix[i][j]);
                    continue;
                }
                flag = false;
            } else {
                if (j > n) {
                    j--;
                    numList.add(matrix[i][j]);
                    continue;
                }
                if (i > n + 1) {
                    i--;
                    numList.add(matrix[i][j]);
                    continue;
                }
                flag = true;
                n++;
                k-=2;
            }
        }
        return numList;
    }
}