59. 螺旋矩阵 II

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 <= n <= 20

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] ans = new int[n][n];
        //设置圈数,判断是否内层是否还有圈是否需要读取（若只剩一个数则退出）
        int loop = 0;
        // 每圈后设置修改初始位置(start, start)(即从内层一圈开始读取)
        int start = 0;
        int count = 1;
        // i代表行数,j代表列数
        int i,j;

        while (loop++ < n/2) {
            for (j = start; j < n - loop; j++) {
                // 这里用start而不用i的原因是第二圈start位置会改变
                ans[start][j] = count++;
            }

            for (i = start; i < n - loop; i++) {
                ans[i][j] = count++;
            }

            for (; j >= loop; j--) {
                ans[i][j] = count++;
            }

            for (; i >= loop; i--) {
                ans[i][j] = count++;
            }
            start++;
        }

        if (n % 2 == 1) {
            ans[start][start] = count;
        }
        return ans;
    }
}