【每日一道算法题】螺旋矩阵II

原题

力扣题目链接(opens new window)

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

示例:

输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]

思路解析

本题并不用到何种算法，更多的是考察对二维数组的理解以及循环时边界条件的确定。

所以本题的解法就是最常规的思路，按照右下左上的顺序进行遍历插入。但是插入时候的边界条件一定要确立清楚，比如在向右插入元素的时候是否包含边界，如果包含边界，那么之后每一次循环遍历也需要包含边界，否则代码实现起来非常混乱。

定义一个变量i表示二维数组的第i行，变量j表示二维数组的第j行。

我的代码

当时一开始看错题了，以为是S形添加，浪费了老半天；最后看了一眼评论区，参考一个思路进行了代码实现。

class Solution {
    public int[][] generateMatrix(int n) {
        //定义一个二维数组，长、宽为n*n，第一个元素固定为1
        int[][] arr = new int[n][n];
        arr[0][0] = 1;

        //定义一个变量i，表示二维数组的第几行；指针j表示二维数组的第几列;num表示要插入数组中的元素，每次插入后进行自增
        int i = 0, j = 0, num = 2;
        if(n == 1){
            return arr;
        }
        //这里表示每次右下左上循环完一圈以后，判断右边是否为0，如果为0，就继续进行右下左上循环；如果不为0，说明添加完了
        while(arr[i][j+1] == 0){
            //右，i不变，j累加，直到j=n-1为止（循环条件j <= n-1），同时判断下一个位置是否为0，num每次插入完自增
            while(j < n-1 && arr[i][j+1] == 0){
                arr[i][j+1] = num++;
                j++;
            }
            //下，j不变，i累加，直到i=n-1为止（循环条件i < n-1），同时判断下一个位置是否为0，num每次插入完自增
            while(i < n-1 && arr[i+1][j] == 0){
                arr[i+1][j] = num++;
                i++;
            }
            //左，i不变，j累减，直到j=0为止（循环条件j>0），同时判断下一个位置是否为0，num每次插入完自增
            while(j > 0 && arr[i][j-1] == 0){
                arr[i][j-1] = num++;
                j--;
            }
            //上，j不变，i累减，直到i=0位置（循环条件i>0），同时判断下一个位置是否为0，num每次插入完自增
            while(i > 0 && arr[i-1][j] == 0){
                arr[i-1][j] = num++;
                i--;
            }
        }
        return arr;
    }
}

优质题解代码解读

这道题解的优点在于边界建立得非常清晰，每次添加元素后就将边界条件内缩即可。

class Solution {
    public int[][] generateMatrix(int n) {
        //定义四个边界条件，分别对应左、右、上、下
        int l = 0, r = n - 1, t = 0, b = n - 1;
        //定义一个二维数组，长宽是n*n
        int[][] mat = new int[n][n];
        //num是要插入的元素值，从1开始，从[0][0]位置开始插入，每次插入后自增，循环条件是num<=n*n
        int num = 1, tar = n * n;
        while(num <= tar){
            //从左边界开始，循环向右添加元素（含右边界）
            for(int i = l; i <= r; i++) mat[t][i] = num++; // left to right.
            //因为右边界已经插入了一个元素，因此上部边界条件t要加1，相当于上部边界向内缩
            t++;
            for(int i = t; i <= b; i++) mat[i][r] = num++; // top to bottom.
            //右边界向内缩
            r--;
            for(int i = r; i >= l; i--) mat[b][i] = num++; // right to left.
            //下边界向上缩
            b--;
            for(int i = b; i >= t; i--) mat[i][l] = num++; // bottom to top.
            //左边界向右缩
            l++;
        }
        return mat;
    }
}