题目描述:
给你一个正整数 n ,生成一个包含 1 到 n^2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
1 <= n <= 20
思路与总结
本题不涉及算法,只是模拟过程,但却十分考察对代码的掌控能力。
求解本题依然要坚持循环不变量原则,类似二分查找中提到的,如果要写出正确的二分法,一定要坚持循环不变量原则。
因此,模拟顺时针画矩阵的过程如下:
填充上行从左到右
填充右列从上到下
填充下行从右到左
填充左列从下到上
【由内向外一圈一圈画下去...】
注意这里的边界条件非常多,需要按照固定规则来遍历,一圈下来需要画四条边,每画一条边都要坚持一致的左闭右开,或者左开右闭的原则。
按照左闭右开的原则,如图所示:每一种颜色代表一条边,遍历的长度可以看出每一个拐角处的处理规则,拐角处让给新的一条边来继续画。
代码示例
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> res(n,vector<int>(n,0));
int startx = 0, starty = 0; //每一圈起始位置
int loop = n / 2; //循环圈数
int count = 1; //每个位置取值
int offset = 1; //每条边的边界间隔数
int mid = n / 2; //中间位置
int i,j;
while(loop--) {
i = startx;
j = starty;
//填充上行从左到右
for(j=starty;j<n-offset;j++) {
res[i][j] = count++;
}
//填充右列从上到下
for(i=startx;i<n-offset;i++) {
res[i][j] = count++;
}
//填充下行从右到左
for(;j>offset-1;j--) {
res[i][j] = count++;
}
//填充左列从下到上
for(;i>offset-1;i--) {
res[i][j] = count++;
}
//更新下一圈的起始位置
startx++;
starty++;
//更新每一圈的间隔数
offset += 1;
}
//n为奇数,需要填充中间位置
if(n % 2) {
res[mid][mid] = count;
}
return res;
}
};