59. 螺旋矩阵 II

59. 螺旋矩阵 II

题目描述

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

示例:

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

---

题解：

与 螺旋矩阵 基本一样，只不过这题是填充元素，按照 螺旋矩阵 的写法填充即可。

写法一：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        if ( !n ) return {};
        const int dx[4] = { 0, 1, 0, -1 };
        const int dy[4] = { 1, 0, -1, 0 };
        int x = 0, y = 0, tx, ty;
        int tot = n * n;
        vector<vector<int>> ret(n, vector<int>(n));
        for ( int i = 1, d = 0; i <= tot; ++i ) {
            ret[x][y] = i;
            tx = x + dx[d];
            ty = y + dy[d];
            if ( tx < 0 || tx >= n || ty < 0 || ty >= n || ret[tx][ty] ) {
                d = ( d + 1 ) % 4;
                x += dx[d];
                y += dy[d];
            } else {
                x = tx;
                y = ty;
            }
        }
        return ret;
    }
};
/*
时间：0ms，击败：100.00%
内存：6.5MB，击败：88.96%
*/

写法二：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        if ( !n ) return {};
        int tot = n * n;
        vector<vector<int>> ret(n, vector<int>(n));
        int k = 1;
        int x = 0, y = 0;
        while ( k <= tot ) {
            while ( y < n && !ret[x][y] ) ret[x][y++] = k++;
            ++x, --y;
            while ( x < n && !ret[x][y] ) ret[x++][y] = k++;
            --x, --y;
            while ( y >= 0 && !ret[x][y] ) ret[x][y--] = k++;
            --x, ++y;
            while ( x >= 0 && !ret[x][y] ) ret[x--][y] = k++;
            ++x, ++y;
        }
        return ret;
    }
};
/*
时间：0ms，击败：100.00%
内存：6.3MB，击败：98.12%
*/