54.螺旋矩阵——学习笔记

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

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

提示：

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 10
• -100 <= matrix[i][j] <= 100

题目来源：力扣（LeetCode）链接

题解：

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> res = new ArrayList<>();//定义一个List集合用来存放结果
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
            //如果matrix为空，就直接返回空的集合
            return res;
        }
        int m = matrix.length;//m表示natrix的行数
        int n = matrix[0].length;//n表示natrix的列数
        int offset = 1;//每圈每行或每列走的步长为m/n-offset
        int startX = 0;//每一圈循环行的起始位置
        int startY = 0;//每一圈循环列的起始位置
        int i,j;//i,j是matrix的索引下标
        if (m == 1) {//说明matrix数组只有一行
            for (int k = 0; k < n; k++) {
                res.add(matrix[0][k]);
            }
            return res;
        }
        if (n == 1) {//说明matrix数组只有一列
            for (int k = 0; k < m; k++) {
                res.add(matrix[k][0]);
            }
            return res;
        }
        while (res.size() < m * n) {
            i = startX;//把起始下标赋给i和j
            j = startY;
            if (res.size() == m * n - 1 && m == n && m % 2 == 1) {
                //处理奇数矩阵的中间位置，如3x3矩阵的[1][1]位置
                int mid = m / 2;
                res.add(matrix[mid][mid]);
                return res;
            }
            //填充上行从左到右(左闭右开)
            for (; j < n - offset; j++) {
                res.add(matrix[i][j]);
                if (res.size() == n * m) {
                    return res;
                }
            }
            //填充右列从上到下(左闭右开)
            for (; i < m - offset; i++) {
                res.add(matrix[i][j]);
                if (res.size() == m * n) {
                    return res;
                }
            }
            //填充下行从右到左(左闭右开)
            for (; j > startY; j--) {
                res.add(matrix[i][j]);
                if (res.size() == m * n) {
                    return res;
                }
            }
            //填充左列从下到上(左闭右开)
            for (; i > startX; i--) {
                res.add(matrix[i][j]);
                if (res.size() == m * n) {
                    return res;
                }
            }
            //每一圈结束后，起始位置都要加1
            startX++;
            startY++;
            //offset控制每圈中每条边的步长
            offset++;
        }
        return res;
    }
}