![7a9cb31cd6161f271346f2c23a25eac4.png](https://i-blog.csdnimg.cn/blog_migrate/c0a3aa9e41c499d5d8af3b1dd9678f2e.jpeg)
二维数组中的查找
题目:
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例:
现有矩阵 matrix 如下:
[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。
给定 target = 20,返回 false。
限制:
0 <= n <= 1000
0 <= m <= 1000
解题思路
首先最先想到的暴力方法是通过遍历整个二维数组,通过每个元素与目标值比对来判断目标值是否在数组中。不过由于题目给定的限制条件是n(0-1000)、m(0-1000),所以考虑到算法的时间复杂度,这种方法不可用。
然后通过题目可知,矩阵的行和列都是递增的,因此我们可以想到从矩阵的右上角的第一个元素开始,通过元素与目标值进行比对,若目标值小于当前元素,那么让当前元素在当前行左移一位;若目标值大于当前元素,那么让当前元素在当前列下移一位;
如下图所示:假设目标值target=16,而右上角第一个元素为15.由于15<16,故让元素从当前列下移一位再进行判断。
![0a13d8b6b29de1b3375b7ba2bf4cd6cc.png](https://i-blog.csdnimg.cn/blog_migrate/0601d8f5a98c92a08ea597894a502871.png)
下移后的效果图如下所示:
![84e218755e60af1974f05c04e20b99b8.png](https://i-blog.csdnimg.cn/blog_migrate/65a4005933d9442b209f69e2680b713a.png)
可以看到下移后19>16,于是让元素从当前行向左移动一位再进行判断。
![ef380f4d637edbda387211b1034f952e.png](https://i-blog.csdnimg.cn/blog_migrate/d363689047632cdf349e196a3a731e34.png)
到这里可以发现12<16,故让元素从当前列下移一位再进行判断。
这个例子可以发现下一步就能找到与目标值匹配的元素,所以返回true;如果通过循环条件判断,当元素移出了矩阵边界还没有找到匹配项,那么返回false,表示矩阵无与目标值匹配到的元素。
代码如下(示例):
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
//判断容器是否为空,若为空则直接返回false
if(matrix.empty())
{
return false;
}
//获取矩阵的行、列
int n=matrix.size();
int m=matrix[0].size();
//从矩阵的右上角第一个元素开始
int i=0,j=m-1;
//判断边界条件
while(i<n && j>=0)
{
//这里通过[]来获取vector容器内的元素(可以把vector嵌套vector看成二维数组)
//如果当前元素的值大于目标值,则向左移动一位再进行判断
if(target<matrix[i][j])
{
j--;
}
//如果当前元素的值小于目标值,则向下移动一位再进行判断
else if(target>matrix[i][j])
{
i++;
}
//若两者相等,则直接返回true
else
{
return true;
}
}
return false;
}
};
执行结果
执行用时:52 ms, 在所有 C++ 提交中击败了94.98%的用户内存消耗:13 MB, 在所有 C++ 提交中击败了86.58%的用户