59.螺旋矩阵II
给你一个正整数 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)); // 使用vector定义一个二维数组
int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
int loop = n / 2; // 每个圈循环的次数
int mid = n / 2; // 矩阵中间的位置,用于处理n为奇数时的情况
int count = 1; // 用来给矩阵中每一个空格赋值,从1开始
int offset = 1; //每次循环缩小边界
int i, j;//用于进行循环
while (loop--) { //进行loop次循环
i = startx;
j = starty;
// 填充上行从左到右(左闭右开)
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
startx++;
starty++;
// offset 控制每一圈里每一条边遍历的长度
offset += 1;
}
// 如果n为奇数的话,需要单独给矩阵最中间的位置赋值
if (n % 2) {
res[mid][mid] = count;
}
return res;
}
};
为了做到不重不漏,就要整体都按照一定的顺序进行赋值。这里就用到了统一使用左闭右开区间的思想,每一轮赋值时统一给左边界赋值,右边界不赋值。
参考教程:一入循环深似海 | LeetCode:59.螺旋矩阵II_哔哩哔哩_bilibili
往期回顾:
LeetCode977.有序数组的平方
LeetCode844.比较含退格的字符串
LeetCode283.移动零
LeetCode27.移除元素
LeetCode26.删除有序数组中的重复项
LeetCode209.长度最小的子数组
LeetCode904. 水果成篮
LeetCode242.有效的字母异位词
LeetCode76.最小覆盖子串