Leetcode.59.螺旋数组Ⅱ
https://leetcode-cn.com/problems/spiral-matrix-ii/
给你一个正整数
n
,生成一个包含
1
到
n2
所有元素,且元素按顺时针顺序螺旋排列的
n x n
正方形矩阵
matrix
。
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
本题思路:本题的关键在于如何得到一个n*n的螺旋数组。其本质就是一个二维数组。然后螺旋式的给这个二维数组赋值,在赋值过程中,要让你的循环条件始终不变,我们采用的是左闭右开的原则,比如上面的数组,我们先添加1,2,再3,4,再5,6 … 。同时考虑根据n的大小设置中间值的大小
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 loop=n/2;//循环圈数;
int mid=n/2;//可能得到的中间值
int count=1;//赋值的初值
int offset=1;//控制赋值的范围值
int i,j;
while(loop--){
i=startx;
j=starty;
for(;j<starty+n-offset;j++){
res[startx][j]=count++;
}
for(;i<startx+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+=2; //因为前后的元素都有值
}
if(n%2){
res[mid][mid]=count;
}
return res;
}
};