LeetCode刷题05-数组-59.螺旋矩阵II

LeetCode刷题05-数组-59.螺旋矩阵II

题目

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

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

示例2：

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

提示：

• 1 <= n <= 20

代码

其难点在于模拟过程，通过模拟可以发现，该螺旋过程可以分解为不断画宽度-1的正方形。

class Solution {
    public int[][] generateMatrix(int n) {
        int num = 1;

        int[][] resultMatrix = new int[n][n];
        int row = 0;
        int column = 0;
        int i = n;
        while(i > 0) {
            //1. 行不增，列增
            while(column < i && num <= n*n) {
                resultMatrix[row][column] = num;
                num++;
                column++;
            }
            column--;
            row++;
            //2. 行增，列不增
            while(row < i && num <= n*n) {
                resultMatrix[row][column] = num;
                num++;
                row++;
            }
            row--;
            column--;
            //3. 行不增，列减
            while(column >= n - i && num <= n*n) {
                resultMatrix[row][column] = num;
                num++;
                column--;
            }
            column++;
            row--;
            //4. 行减，列不增
            while(row > n - i && num <= n*n) {
                resultMatrix[row][column] = num;
                num++;
                row--;
            }
            row++;
            column++;
            i--;
        }
        return resultMatrix;
    }
}