class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
int a[n+2][n+2];
memset(a,0,sizeof(a));
for(int i=0;i<n+2;i++){
a[0][i]=1;
a[i][0]=1;
a[n+1][i]=1;
a[i][n+1]=1;
}
int i=1,j=1;
int d=0;
int num=1;
vector<vector<int> > b(n,vector<int>(n,0));
while(1){
if(!a[i][j])
a[i][j]=num++;
else
break;
if(d==0){
++j;
if(a[i][j]){
--j;
d=(d+1)%4;
++i;
}
}
else if(d==1){
++i;
if(a[i][j]){
--i;
d=(d+1)%4;
--j;
}
}
else if(d==2){
--j;
if(a[i][j]){
++j;
d=(d+1)%4;
--i;
}
}
else{
--i;
if(a[i][j]){
++i;
d=(d+1)%4;
++j;
}
}
}
for(int p=1;p<n+1;++p)
{
for(int q=1;q<n+1;++q){
b[p-1][q-1]=a[p][q];
}
}
return b;
}
};
题目大意:给出一个数n,往二维向量中填入1~n*n个数,填数方式为螺旋形方式填数。
算法思想:
1.构造一个(n+2)*(n+2)规模的数组,初始化为0.
2.将上述数组四周边界填充为1.
3.按照顺时针方向往数组中填数,共有四个方向:0,1,2,3,4
1)0 时向右移动,如果遇到边界,则撤回并向下移动。
2)1时向下移动,如果遇到边界,则撤回并向左移动。
3)2时向左移动,如果遇到边界,则撤回并向上移动。
4)3时向上移动。如果遇到边界,则撤回并向右移动。
4.当没有可填位置时,数组填完。每填写一个数字,其数值增1.
5.将数组中的数装入二维向量中。
6.返回该向量。
代码如下: