力扣59. 螺旋矩阵 II Java版本

---

思路

先把数组都置为零，然后一个个填数

题目

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

示例 1：

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

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

提示：

1 <= n <= 20

解题方法

先把所有的数都置为零，然后按照循环的顺序依次填入数字。
以向右为例：当加一之后没有超出数组的边界并且该位置的数为0，则证明可以继续填数。一旦加一超出数组边界或者已经填写过，就开始考虑向下，然后再依次考虑向左、向上

代码

class Solution {
    public int[][] generateMatrix(int n) {
        /*把所有的与元素都设置为0，如果右侧的数为0就往右走，如果下面的数为0就往下走，向左向上也类似*/
        int[][] a = new int[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                a[i][j] = 0;
            }
        }
        int i = 0;//表示行
        int j = 0;//表示列
        int num = 1;//用来计数，填写矩阵
        a[0][0] = num++;//先把第一个位置赋值，因为有可能n=1，后边直接不用管了
        if(n==1){//直接返回就可以了
            return a;
        }
        while (true) {
            //向右
            while ((j + 1) < n && a[i][j + 1] == 0) {
                j++;
                a[i][j] = num++;
            }
            //向下
            while ((i + 1) < n && a[i+1][j] == 0) {
                i++;
                a[i][j] = num++;
            }
            //向左
            while ((j - 1) >= 0 && a[i][j - 1] == 0) {
                j--;
                a[i][j] = num++;
            }
            //向上
            while ((i - 1) >= 0 && a[i-1][j] == 0) {
                i--;
                a[i][j] = num++;
            }
            //如果num==(n*n+1)则证明已经把所有的位置都填上数了
            //因为这里采用的是num++而不是++num，所以是num==(n*n+1)而不是num==n*n
            if(num==(n*n+1)){
                return a;
            }

        }

    }
}