题源: 59. 螺旋矩阵 II
思路
- 确定外围圈数:
int loop = n/2; //螺旋圈数
当 n 为偶数时,例如 n = 4,那么最外层是一个 4x4 的框架,里面是一个 2x2 的框架。这里共有两层,也就是两圈。
当 n 为奇数时,例如 n = 5,那么最外层是一个 5x5 的框架,里面依次是 3x3,最后是 1x1(只处理中心点)。这里共有 (5/2) = 2 圈外围,最后的单一中心点单独处理。 - 层次迭代:
通过迭代每一层,你从外圈开始填充,每次处理完一圈后,实际处理的区域越来越小,直到完成整个矩阵的填充。
每次处理一圈后,下一圈的起始位置向内缩进一行和一列,大小也相应减小。 - 处理剩余中心:
对于奇数 n,n / 2 表示完整圈数。如果 n 是奇数,矩阵的中心将是一个单独的点,位于 (n/2, n/2)。这个中心点不属于任何圈,需要在循环之外单独处理。 - 代码大体流程
在螺旋矩阵的填充算法中,loop = n / 2 的实际意义是你需要完全遍历的层数或圈数。每一“圈”都包括四个方向的遍历(向右、向下、向左、向上),每遍历完一圈,就从外向内收缩到下一个更小的圈进行相同的操作,直至处理完所有圈。如果 n 是奇数,处理剩下的中心元素。
复杂度
-
时间复杂度 O(n^2): 模拟遍历二维矩阵的时间
-
空间复杂度 O(1)
Code
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> res(n, vector<int> (n, 0));
int loop = n/2; //螺旋圈数
int startx = 0, starty = 0, mid = n/2; //最中间的数
int offset = 1;
int cnt = 1;
int i, j;
while(loop --){
i = startx, j = starty;
//上行,从左到右
for(j;j < n - offset;j ++)
res[i][j] = cnt ++;
//右列,从上到下
for(i;i < n - offset;i ++)
res[i][j] = cnt ++;
//下行,从右到左
for(j;j > startx;j --)
res[i][j] = cnt ++;
//左列,从下到上
for(i;i > starty;i --)
res[i][j] = cnt ++;
//更新各个条件
startx ++, starty ++, offset ++;
}
if(n % 2) res[mid][mid] = cnt; //若n是奇数,填充最中间的数字
return res;
}
};