给你一个正整数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> v(n,vector<int>(n));
vector<vector<int>> d={{0,1},{1,0},{0,-1},{-1,0}};
int cur=1;
int row=0,col=0;
int dindex=0;
while(cur<=n*n){
v[row][col]=cur;
int nextrow=row+d[dindex][0];
int nextcol=col+d[dindex][1];
if(nextrow<0||nextrow>=n||nextcol<0||nextcol>=n||v[nextrow][nextcol]!=0){
dindex=(dindex+1)%4;
}
row=row+d[dindex][0];
col=col+d[dindex][1];
cur++;
}
return v;
}
};