剑指 Offer 04. 两种方法通过二维数组中的查找
一、题目:
在一个 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
二、思路、代码和结果:
其实这道题应该归到简单题里面去。
方法一:暴力搜索
这没啥道理,就是把二维数组中的每个值都挨个遍历一次,有相同的就返回true,没有就返回false;
n*m还挺小的,没超时,乌拉!
//暴力搜索
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
for(int i=0;i<matrix.size();i++){
for(int j=0;j<matrix[0].size();j++){
if(matrix[i][j]==target)return true;
}
}
return false;
}
};
结果:
但是这似乎不太高效,因为要把所有的值都遍历一次,更适合乱序的二维数组。
但是!
题目中的矩阵是有一定的顺序的:每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。
我们可以利用这一特性对方法一进行改进。于是就有了:
方法二:利用所给顺序特性遍历
依据题目所给矩阵每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序的特性。我们可以从右上角开始遍历至左下角的位置。
具体做法:
定义行row和列col的变量,因为从右上角开始,所以行row和列col分别初始化为0和matrix[0].size()-1;
我们看第row行,如果最后一个数字大于target,那说明第row行不可能有这个数字,但是下一行可能有,所以行row++;
如果第row行最后一个数字小于等于target,那说明第row行可能(小于的时候可能,等于的时候肯定)有这个数字,从后往前倒着遍历第row行,列col–,和target对比有相同的就返回true,没有就返回false;
完美~
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
if(matrix.size() == 0) return false;
int row=0,col=matrix[0].size()-1;
while(row<matrix.size() && col>=0){
if(matrix[row][col]>target)col--;
else if(matrix[row][col]<target) row++;
else
return true;
}
return false;
}
};
对比:
第二种方法少了一层循环,复杂度大大降低,从提交的时间和空间来看也是低了很多,很好啊!
如有不足,欢迎指正!
怕什么真理无穷,进一寸有进一寸的欢喜!