leetcode 59
链接:力扣 。
题目:
给定一个正整数 n,生成一个包含 1 到 n^2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
思路:
模拟顺时针画矩阵的过程:
- 填充上行从左到右
- 填充右列从上到下
- 填充下行从右到左
- 填充左列从下到上
每次一圈下来,要画每四条边,在画每条边的时候都要坚持一致的左闭右开原则,就可以将矩阵的最外圈填充完毕。对于输入为 n 的情况下,需要 圈。但要注意到,在n为奇数时最后一圈只有中心一个元素了,这种情况需要单独处理。因此我们将循环次数定为
,如果n为奇数我们再单独处理中心元素。
参考代码:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int> > res(n, vector<int>(n, 0));
int loop = n / 2;
int count = 1; //计数器
int offset = 0; //偏移值,代表已经填充完offset个最外圈
int i, j;
while (loop--) {
//从左至右填充上边(左闭右开)
for (j = offset; j < n -1 - offset; j++) {
res[offset][j] = count++;
}
//从上至下填充上边(上闭下开)
for (i = offset; i < n -1 - offset; i++) {
res[i][j] = count++;
}
//从右至左填充下边(右闭左开)
for (; j > offset; j--) {
res[i][j] = count++;
}
//从下至上填充左边(下闭上开)
for (; i > offset; i--) {
res[i][j] = count++;
}
//填充完一圈,偏移加一
offset++;
}
//如果n是奇数,单独处理中点
if (n % 2) {
res[n / 2][n / 2] = count;
}
return res;
}
};