给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
解析:模拟填充即可
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
if(n==0) return {{}};
vector<vector<int> > res(n,vector<int>(n,0));
int down=n-1,up=0,left=0,right=n-1;
int count=1;
while(left<=right && up<=down){
for(int i=left;i<=right;i++) res[up][i]=count++;
for(int i=up+1;i<=down;i++) res[i][right]=count++;
for(int i=right-1;i>=left;i--) res[down][i]=count++;
for(int i=down-1;i>=up+1;i--) res[i][left]=count++;
++up,--down;
++left,--right;
}
return res;
}
};