LeetCode 每日一题59. 螺旋矩阵 II

59. 螺旋矩阵 II

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

示例 1：

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

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

提示：

• 1 <= n <= 20

方法一：模拟

解题思路

• 题目 [54. 螺旋矩阵] 的逆过程，依然采用模拟的方式。
• 从最外圈一圈一圈地往中心逼近，对于每一圈，依次从 [上 -> 右 -> 下 -> 左] 的顺序添加元素。

参考代码

public int[][] generateMatrix(int n) {
    int[][] matrix = new int[n][n];
    int num = 1;
    int top = 0, bot = n - 1, left = 0, right = n - 1;
    while (top <= bot && left <= right) {
        int l = left;
        while (l <= right) {
            matrix[top][l] = num++;
            l++;
        }
        int t = top + 1;
        while (t <= bot) {
            matrix[t][right] = num++;
            t++;
        }
        if (top < bot && left < right) {
            int r = right - 1;
            while (r >= left) {
                matrix[bot][r] = num++;
                r--;
            }
            int b = bot - 1;
            while (b > top) {
                matrix[b][left] = num++;
                b--;
            }
        }
        top++;
        bot--;
        left++;
        right--;
    }
    return matrix;
}

执行结果