题目
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
1 <= n <= 20
解题思路
每一圈对应四条边,每条边的起点和终点都不是固定的,同时对于每一边我们所采用的策略就是左闭右开,不包含最后一个点,通过四条边之后,所有的点都被揽括其中
输入参数n,就意味着平方之后就变成了一个n*n的矩阵,每 循环一圈之后,对于行来说,就会减少到两行所以循环的深度就是n/2,
对于+x方向的第一个循环,第二圈的时候起点就变了,所以直接用一个变量来操作即可,每次循环的时候为这个变量递增,
遍历+x反向之后,就是遍历-y方向,也是从startx开始遍历,遍历的终点式n-offset中
之后就是遍历-x方向,在遍历-x方向的时候,起点式j在+x的时候已经达到了目标点,所以不需要对其赋初值,循环结束的条件的就是大于starty
最后就是遍历+y方向,+y方向怎么取值,i在-y的时候已经达到了目标点,只是还没有给它赋值而已,所以遍历的终止条件就是i>startx
循环结束之后,对startx,starty递增,同时对offset递增
还需要考虑的一个点就是,如果所给的这个数式奇数的话,式需要单独考虑的,是奇数的时候,中间的点不会被赋值,需要手动对最中心的位置赋值为count。
代码(C++)
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> res(n,vector<int>(n,0));//使用vector定义一个二维数组
int startx=0,starty=0;//每一圈的起止点
int offset=1;//每一圈中的重点
int loop=n/2;
int mid=n/2;
int count=1;
int i,j;
if(n==1){
res[0][0]=1;
return res;
}
while(loop--){
i=startx;
j=starty;
for(j=starty;j<n-offset;j++){
res[startx][j]=count++;
}
for(i=startx;i<n-offset;i++){
res[i][j]=count++;
}
for(;j>starty;j--){
res[i][j]=count++;
}
for(;i>startx;i--){
res[i][j]=count++;
}
startx++;
starty++;
offset+=1;
if(n%2) res[mid][mid]=count;
}
return res;
}
};