59.螺旋矩阵
注意点
边界条件非常多,处理边界容易出问题
和二分法一样,需要遵循循环不变量的原则
不变量:对每一条边的处理规则,每一条边保持同样的左闭右开/左闭右闭
方法一:边界逐步缩短
题解链接:Spiral Matrix II (模拟法,设定边界,代码简短清晰) - 螺旋矩阵 II - 力扣(LeetCode)
public int[][] generateMatrix(int n) {
//大神解法的解读
int left = 0, right = n-1, top = 0, bottom = n-1;
int count = 1, target = n * n;
int[][] res = new int[n][n];
//for循环中变量定义成i或j的细节:按照通常的思维,i代表行,j代表列
//这样,就可以很容易区分出来变化的量应该放在[][]的第一个还是第二个
//对于变量的边界怎么定义:
//从左向右填充:填充的列肯定在[left,right]区间
//从上向下填充:填充的行肯定在[top,bottom]区间
//从右向左填充:填充的列肯定在[right,left]区间
//从下向上填充:填充的行肯定在[bootom,top]区间
//通过上面的总结会发现边界的起始和结束与方向是对应的
while(count <= target){
//从左到右填充,相当于缩小上边界
for(int j = left; j <= right; j++) res[top][j] = count++;
//缩小上边界
top++;
//从上向下填充,相当于缩小右边界
for(int i = top; i <=bottom; i++) res[i][right] = count++;
//缩小右边界
right--;
//从右向左填充,相当于缩小下边界
for(int j = right; j >= left; j--) res[bottom][j] = count++;
//缩小下边界
bottom--;
//从下向上填充,相当于缩小左边界
for(int i = bottom; i >= top; i--) res[i][left] = count++;
//缩小左边界
left++;
}
return res;
}
按照通常的思维,i代表行,j代表列
1.从左到右填充,缩小上边界
2.从上到下填充,缩小右边界
3.从右向左填充,缩小下边界
4.从下向上填充,缩小左边界
注意点:
- 使用
num <= tar
而不是l < r || t < b
作为迭代条件,是为了解决当n
为奇数时,矩阵中心数字无法在迭代过程中被填充的问题。(此问题可以画图或者VS中进行debug尝试) - 当n=3的时候,如果循环条件是
l < r || t < b
,此时会出现螺旋矩阵中间的数值空缺的问题,原因是当转到[1][1]
的时候,都不满足上述条件,中间的值就不会被填充。 - 但,条件只要改为
l <= r || t <= b
,即可。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
int num = 1;
int left = 0, top = 0, right = n - 1, bottom = n - 1;
//初始化数组
vector<vector<int>> res(n,vector<int>(n));
while (left <= right || top <= bottom ) {
//left to right
for (int i = left; i <= right; ++i) res[top][i] = num++;
++top;
//top to bottom
for (int i = top; i <= bottom; ++i) res[i][right] = num++;
--right;
//right to left
for (int i = right; i >= left; --i) res[bottom][i] = num++;
--bottom;
//bottom to top
for (int i = bottom; i >= top; --i) res[i][left] = num++;
++left;
}
return res;
}
};
方法二:判断方向,在到了边界的时候修改方向
题解链接:官方解答的评论区
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> ans(n, vector<int>(n));
int t = 0; // 初始方向
int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}; // 四个方向
for (int i = 0, j = 0, k = 1; k <= n * n; ++k) {
ans[i][j] = k;
int x = i + dx[t], y = j + dy[t]; // 计算下个方向
// 判断下个点是否合理,不合理就改方向
if (x < 0 || x >= n || y < 0 || y >= n || ans[x][y] != 0)
t = (t + 1) % 4;
i += dx[t];
j += dy[t];
}
return ans;
}
};
注意点:
1.如何使用vector容器定义二维数组?
在 C++ 中创建一个二维向量(实际上是一个向量的向量)。
vector<int>(n)
创建了一个大小为 n 的整数向量,所有元素都会被初始化为 0。
vector<vector<int>>(n, vector<int>(n))
则创建了一个包含 n 个 vector<int>(n)
的向量。换句话说,它创建了一个 n x n 的二维数组,所有元素都被初始化为 0。
所以,vector<vector<int>> ans(n, vector<int>(n));
创建了一个名为 ans 的 n x n 的二维整数数组,所有元素都初始化为 0。这可以用于存储二维数据结构,如矩阵。
2.t = (t + 1) % 4
含义?
t = (t + 1) % 4
,该公式中t从0开始取值,公式的结果是0,1,2,3这四个数字的循环(在t初始值为0的情况下)。该公式用于取值包含四个元素的数组,本题目中变量t用于更改偏移量,从而确定方向。
3.如何通过dx
和dy
的取值改变移动方向?
dx
和dy
的取值是更改方向的重要因素。
dx
和dy
数组的定义对应了在二维矩阵中向四个方向(右,下,左,上)移动的坐标变化规则。这四个方向是按照螺旋矩阵的生成规则设定的。
dx
数组对应的是行的偏移量。在二维矩阵中,行的增减对应的是上下移动。所以dx
数组中的1
和-1
分别表示向下和向上移动一格,0
表示在行上不移动。dy
数组对应的是列的偏移量。在二维矩阵中,列的增减对应的是左右移动。所以dy
数组中的1
和-1
分别表示向右和向左移动一格,0
表示在列上不移动。
以**初始的方向t=0
为例,此时dx[0]
和dy[0]
分别为0
和1
,这表示初始的移动方向为向右。**当在矩阵的边界或者已经填充的单元格需要改变方向时,通过t = (t + 1) % 4
来改变dx
和dy
的索引,也就改变了方向。例如,当t=1
时,dx[1]
和dy[1]
分别为1
和0
,这表示当前的移动方向为向下。
因此,dx
和dy
的定义是dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}
。通过这种方式,可以便捷地控制在二维矩阵中的移动方向。
总结为:
向右移动→:t=0, dx[0] = 0, dy[0] = 1;
向下移动↓: t=1, dx[1] = 1, dy[1] = 0;
向左移动←: t=2, dx[2] = 0, dy[1] = -1;
向上移动→: t=3, dx[3] = -1, dy[1] = 0;
4.判断是否需要换方向的边界条件,为什么是x >= n
而不是x>n
?
在计算机编程中,数组的索引通常是从0开始的。所以对于一个大小为n的数组,其有效索引的范围是0到n-1。
当我们检查x
或y
是否超出了矩阵的边界时,我们需要比较它们是否等于或者大于n
。如果x
或y
等于n
,那么它们已经超出了数组的有效索引范围,因为最大的有效索引是n-1
。
所以我们使用x >= n
而不是x > n
来判断x
是否超出了数组的边界。同理,对于y
也是如此。如果我们使用x > n
作为判断条件,那么当x
等于n
时,我们的代码可能会试图访问数组的越界索引,从而可能导致运行时错误。
取余操作注意
对n进行%运算的话,得到的结果一定<n!
本题t=(t+1)%4,操作目的就是按顺序取所有<4的数组下标!
我们可以知道,如果我们想按照顺序取所有<n的正整数,我们也可以使用t=(t+1)%n (t初值赋值为t=0)这种技巧,因为取模%运算得到的一定是<n的数字!