LeetCode #59 螺旋矩阵II——每日一题

59. 螺旋矩阵 II

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

示例 1：

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

示例 2：

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

提示：

• 1 <= n <= 20

和上题反向操作一波就行，要注意的是不用push_back赋值的话需要先开辟空间

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int shang = 0, xia = n, zuo = 0, you = n;
        int len = 0;
        vector<vector<int> > matrix(n);
        for(int i = 0;i < n;i++) matrix[i].resize(n);
        while(1){
            for(int i = zuo;i < you;i++) matrix[shang][i]=++len;
            if(len == n*n) break;
            shang++;
            for(int i = shang;i < xia;i++) matrix[i][you-1]=++len;
            if(len == n*n) break;
            you--;
            for(int i = you-1;i >= zuo;i--) matrix[xia-1][i]=++len;
            if(len == n*n) break;
            xia--;
            for(int i = xia-1;i >= shang;i--) matrix[i][zuo]=++len;
            if(len == n*n) break;
            zuo++;
        }
        return matrix;
    }
};