【代码随想录 | 数组 05】螺旋矩阵 ||

5.螺旋矩阵2

5.1题目

59. 螺旋矩阵 II

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

• 示例一：

  

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

• 示例二：

输入：n = 1
输出：[[1]]

5.2思路

• 每次赋值可分为4次：上、右、下、左

• 首先明确每次赋值，确保都是==左闭右开==（即1 2；3 4；5 6；7 8）

• 循环的圈数：loop=n/2；n为3时循环一圈；n为4时循环两圈
• 若 n为奇数，则最后中间还有个值，arr(n/2)(n/2)
• 每次循环过后，右边界都要缩小一格
• 注意每次循环过后，起始点的位置！！！

class Solution {
    public int[][] generateMatrix(int n) {
        
        int loop=n/2;
        int count=1;
        int[][] arr=new int[n][n];
        int start=0;                //每次循环的开始点（start,start）
        int i=0;
        int j=0;
        int offset=1;               //每次循环右边界收缩一位，控制每条边的遍历长度
        for(int k=0;k<loop;k++){    //k表示正在循环的圈数

            i=start;                //更新起始点
            j=start;

            //上：从左到右
            for(j=start;j<n-offset;j++){
                arr[i][j]=count++;
            }

            //右：从上到下
            for(i=start;i<n-offset;i++){
                arr[i][j]=count++;
            }

            //下：从右到左
            for(;j>start;j--){
                arr[i][j]=count++;
            }

            //左：从下到上
            for(;i>start;i--){
                arr[i][j]=count++;
            }

            start++;    //循环起始点加一
            offset++;
        }
        if (n % 2 == 1) {
            arr[n/2][n/2] = count;
        }
        return arr;
    }
}