59. 螺旋矩阵 II

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

示例 1：

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

示例 2：

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

思路

同54. 螺旋矩阵

1. 边界控制

   ：我们使用四个变量来控制当前遍历的边界：

   • up: 当前遍历的最上面一行。
   • down: 当前遍历的最下面一行。
   • left: 当前遍历的最左边一列。
   • right: 当前遍历的最右边一列。

2. 螺旋遍历：我们按顺时针方向，从左到右、从上到下、从右到左、从下到上 依次遍历当前边界范围内的矩阵元素，并更新边界值。直到所有元素都被遍历。

3. 使用额外的一个变量 index，填充结果矩阵。

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] res = new int[n][n];
        int up = 0, down = n-1, left = 0, right = n-1;
        int index = 1;
        while(true){
            for(int i = left; i <= right; i++){
                res[up][i] = index++;
            }
            if(++up > down){
                break;
            }
            for(int i = up; i <= down; i++){
                res[i][right] = index++;
            }
            if(--right < left){
                break;
            }
            for(int i = right; i >= left; i--){
                res[down][i] = index++;
            }
            if(--down < up){
                break;
            }
            for(int i = down; i >= up; i--){
                res[i][left] = index++;
            }
            if(++left > right){
                break;
            }
        }
        return res;
    }
}