题目内容

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

示例

示例 1：

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

示例 2：

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

提示

1 <= n <= 20

题解

本题和lc54思路一致，不同之处在于本题是需要填入数字。

在方法上，我们定义按照顺时针的顺序排布的方向数组，然后进行遍历。这里仅需要注意一下边界问题就好，即给出的元素坐标都应该在[0,n]以内，且保证格子中没有填入时，就填入当前遍历的值。

具体，请看代码。

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n));
        int dx[] = {0,1,0,-1}, dy[] = {1,0,-1,0};
        for(int i = 1, x = 0, y = 0, d = 0; i <= n * n; i++){
            res[x][y] = i;
            int a = x + dx[d], b = y + dy[d];
            if(a >= n || a < 0 || b >= n || b < 0 || res[a][b]){
                d = (d + 1) % 4;
                a = x + dx[d], b = y + dy[d];
            }
            x = a, y = b;
        }
        return res;
    }
};