力扣59
题目描述:给你一个正整数 n
,生成一个包含 1
到 n2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
循环不变量
在本题中,如果每一次都采用起始位置填充,结束位置也填充的方法,那么情况会变得多样且复杂,所以我们采用循环不变量,对于每一次的填充,我们只填充结束位置之前的空位,而最后一位作为下一次的起始位置来填充。
每一轮(每一圈)作为一次循环,所以我们需要记录需要多少圈,也需要一个标志来判断现在是哪一圈,所以我们自然而然的想到来初始化一个loop作为计数,我们总共需要的轮数也就是2/n。
每一次的填充要么是横向,要么是纵向,所以一定会有一个起始的x与y值,这里我么采用startx 与starty,初始位置为左上角,也就是(0,0),每一轮结束之后,向右下转移,也就是startx++,starty++,此时也应当loop++。
我们知道了起始位置,那么每一轮需要的步长也应当定义一下,自然而然的可以得出,每一轮的步长应当与轮数loop有关,比如第一轮的时候,每一次的步长应该为n-当前的轮数。所以定义一个offset,每一次的遍历到达n-offset即可停止。
思路
首先我们需要初始化startx,starty为(0,0),然后将loop设置为n/2,代表总共需要loop轮的填数,当loop为0时,代表已经填数完毕,可以返回。然后将offset设置为1,每轮(四次填数)后+1,代表外圈已经填充完毕,然后loop--,将初始位置转移到右下角一格,这样一轮也就完成了。
所以代码的结构显而易见,首先一轮循环中是有四个小部分的,所以一定是一个while里面包含了四个小的操作循环,而循环结束的条件也就是loop==0。
代码
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; // 每个圈循环几次,例如n为奇数3,那么loop = 1 只是循环一圈,矩阵中间的值需要单独处理
int mid = n / 2; // 矩阵中间的位置,例如:n为3, 中间的位置就是(1,1),n为5,中间位置为(2, 2)
int count = 1; // 用来给矩阵中每一个空格赋值
int offset = 1; // 需要控制每一条边遍历的长度,每次循环右边界收缩一位
int i,j;
while (loop --) {
i = startx;
j = starty;
// 下面开始的四个for就是模拟转了一圈
// 模拟填充上行从左到右(左闭右开)
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++;
}
// 第二圈开始的时候,起始位置要各自加1, 例如:第一圈起始位置是(0, 0),第二圈起始位置是(1, 1)
startx++;
starty++;
// offset 控制每一圈里每一条边遍历的长度
offset += 1;
}
// 如果n为奇数的话,需要单独给矩阵最中间的位置赋值
if (n % 2) {
res[mid][mid] = count;
}
return res;
}
};