LeetCode 59. 螺旋矩阵 II

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

示例 1：

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

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

总体思路和螺旋矩阵1相似，但是需要自己用数填充矩阵，用一个count计数非常棒！！

class Solution {
    public int[][] generateMatrix(int n) {
       int res[][] = new int[n][n];//结果数组
        int flag[][] = new int[n][n];//1表示取过，0表示未娶
        int i=0;//行
        int j=0;//列.
        int count=0;//当前添加的个数，个数与数组的数值是对应的
        while(count < n*n){

            //右
            while(j<n && flag[i][j]==0){
                count++;//个数+1
                res[i][j] = count;//赋值
                flag[i][j]=1;
                j++;
            }
            j--;
            i++;

            //下
           while(i<n && flag[i][j]==0){
                count++;
                res[i][j]=count;
                flag[i][j]=1;
                i++;
            }
            i--;
            j--;

            //左
            while(j>=0 && flag[i][j]==0){
                count++;
                res[i][j]=count;
                flag[i][j]=1;
                j--;
            }
            j++;
            i--;

            //上
            while(i>=0 && flag[i][j]==0 ){
                count++;
                res[i][j]=count;
                flag[i][j]=1;
                i--;
            }
            i++;
            j++;
        }
        return res;
    }
}