1题目要求
给你一个正整数 n
,生成一个包含 1
到 n2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
2.代码 (C++)
class solution {
public:
vector <vector <int>> generateMatrix (int n) {
vector <vector <int>> res ( n, vector <int> (n,0));
int loop = n/2; //用来记录填满矩阵需要多少圈
int i, j;
int mid = n/2; //用来记录中间元素的下标,因为矩阵根据矩阵大小奇偶的不同,中间的元素我们可能需要手动填上
int startx = 0;//每一次循环开始,准备填矩阵的开始位置
int starty = 0;
int offset = 1; // 每次填数字时,圈的大小都会变化,即n-offset
int count = 1;// 要填入的数值
while(loop--){
i = startx;
j = starty;
// 填圈的区间都按照左闭右开,顺时针方向进行
for(j; j < n-offset; j++){
res[i][j] = count++;
}
for(i; 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;
}
};
3.思路
此题的关键在于确认好边界条件,我用的是左闭右开。
在看到题目时,首先可以确定用循环来解决。那么我们就必须知道循环的圈数loop,而在确定圈数的过程中,会很自然地发现,矩阵大小n的奇偶会需要分开处理。因为n为奇数时,最中央的元素res[mid][mid]是无法通过循环的圈数填好的,所以我们需要自己手动填充。
接着就要确定循环的过程,每次循环时都要确定好开始位置,而其位置是会随着循环的次数而改变的,一开始开始位置下标都为0,随着循环次数递增。
然后每一次填数字时,除了开始位置,我们还需要知道什么时候就算填完这一行(这一列)了,因此有了变量n-offset。
因为是左闭右开的边界限制,所以我们的边界条件不需要算入等于的情况。