算法-螺旋矩阵

1、题目来源

54. 螺旋矩阵 - 力扣（LeetCode）

2、题目描述

给你一个 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

3、题解分享

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        // 思路：一圈一圈的打印 + 固定住每一圈的左上角和右下角
        int n = matrix.length;
        int m = matrix[0].length;
        int row1 = 0,col1 = 0;
        int row2 = n - 1, col2 = m- 1;
        List<Integer>  ans = new ArrayList<>();
        while(row1 <= row2 && col1 <= col2){
            spiralPrint(matrix,ans,row1++,col1++,row2--,col2--);
        }
        return ans;
    }
    public void spiralPrint(int[][] matrix,List<Integer> ans,int row1,int col1,int row2,int col2){
        if(row1 == row2){//只有一行
            for(int i = col1;i<=col2;++i){
                ans.add(matrix[row1][i]);
            }
        }else if(col1 == col2){//只有一列
            for(int i = row1;i<=row2;++i){
                ans.add(matrix[i][col1]);
            }
        }else{
            // 打印上边
            for(int i = col1;i<col2;++i){
                ans.add(matrix[row1][i]);
            }
            // 打印右边
            for(int i = row1;i<row2;++i){
                ans.add(matrix[i][col2]);
            }
            // 打印下边
            for(int i = col2;i>col1;--i){
                ans.add(matrix[row2][i]);
            }
            // 打印左边
            for(int i = row2;i> row1;--i){
                ans.add(matrix[i][col1]);
            }
        }
    }
}