螺旋矩阵 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解答

看这一张图，我们可能很难发现其中的规律，可以尝试画一下 4 阶或者 5 阶矩阵，我们可以发现，螺旋排列的过程可以看作不断走完圆圈的过程，例如 4 阶矩阵可以看作走了两圈，5 阶矩阵可以看作走了三圈，第三圈只有一个元素。
每一圈的步骤是相似的，我们可以把一圈拆分成 4 条边，每条边的长度为阶数 n - 1。
并且我们注意到，每一圈的起始位置为（0，0），（1，1）……

public class GenerateMatrix {

    public static void main(String[] args) {
        int n = 3;
        System.out.println(generateMatrix(n));
    }

    public static int[][] generateMatrix(int n) {
        int[][] nums = new int[n][n]; // 新建一个 n 阶数组
        int mid = n / 2;              // 数组中心位置
        int circles = n / 2;          // 循环的圈数
        int count = 1;                // 递增数，为数组赋值
        int cir = 1;                  // 记录目前圈数

        while (circles-- > 0) {
            // i j 为数组下标，记录每圈的起始位置
            int i = cir - 1, j = cir - 1;
            for (; j < n - cir; j++) nums[i][j] = count++;
            for (; i < n - cir; i++) nums[i][j] = count++;
            for (; j > cir - 1; j--) nums[i][j] = count++;
            for (; i > cir - 1; i--) nums[i][j] = count++;
            cir++;
        }
        // 奇数时，最后一圈只有一个元素
        if (n % 2 == 1) nums[mid][mid] = count;
        return nums;
    }
}