LeetCode-59. 螺旋矩阵 II

题目:

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

示例:

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

解题思路：

设置这样四个变量：

a：左上角，往数组右边走(行),

b：左上角，往数组下边走（列）,

c：右下角，往数组上边走(行)，

d：右下角，往数组左边走(列),

这样就形成了一个矩形先遍历a行b到d-1列，然后遍历d列a到c-1行，然后是c行d到b-1列，最后是b列c到a-1行，，，这样类推，遍历完外层,再就将四脚的坐标向内的对角线移动，再按照上面的规则继续遍历，在遍历的过程中同时给数组赋值即可。

代码实现:

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] ans = new int[n][n];
        int a = 0;//行
        int b = 0;//列
        int c = n-1;//行
        int d = n-1;//列
        int k = 1;
        while(a <= c && b <= d){
            if(a == c){//只有一行
                for(int i=b;i<=d;i++){
                ans[a][i] = k++;
                }
               break;
            }
            if(b == d){//只有一列
                for(int i=a;i<=c;i++){
                    ans[i][d] = k++;
                }
                break;
            }
            for(int i=b;i<d;i++){//首行
                    ans[a][i] = k++;
            }
            for(int i=a;i<c;i++){//尾列
                    ans[i][d] = k++;
            }
            for(int i=d;i>b;i--){//尾行
                    ans[c][i] = k++;
            }
            for(int i=c;i>a;i--){//首列
                    ans[i][b] = k++;
            }
            ++a;
            ++b;
            --c;
            --d;
        }
        return ans;

    }
}

复杂度分析:

• 时间复杂度：$O(n^2)$
• 空间复杂度：$O(n^2)$