坚持循环不变量的原则。 模拟顺时针画矩阵的过程。
一圈下来,画四条边,画的每条边都要坚持一致的左闭右开或左开右闭,这样一圈才能安装统一的规则画下来。
这里每一种颜色,代表一条边,我们遍历的长度,可以看出每一个拐角处的处理规则,拐角处让给新的一条边来继续画。
代码:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> res(n, vector<int>(n, 0));
int startx = 0, starty = 0, i, j; //定义每循环一圈的起始位置
int loop = n / 2; //每个圈循环几次,如若n = 3,则loop = 1,矩阵中间的位置需单独处理
int offset = 1; //区间大小为n - offset;
int cnt = 1;
while(loop--){
i = startx, j = starty;
//区间左闭右开[starty, n + startx - offset) 长度 n - offset
for(j = starty; j < n + starty - offset; j++){
res[i][j] = cnt;
cnt++;
}
for(i = startx; i < n + starty- offset; i++){
res[i][j] = cnt; //[startx, n + startx - offset) 长度 n - offset
cnt++;
}
for(; j > starty; j--){ //
res[i][j] = cnt;
cnt++;
}
for(; i > startx; i--){
res[i][j] = cnt;
cnt++;
}
offset += 2;
starty += 1;
startx += 1;
}
if(n % 2 != 0) res[starty][starty] = cnt;
return res;
}
};