1、题目
就是给定一个二维数组,数组里面从左到右,从上到下都是非降序。然后查找某个值是否存在。这个题目在剑指offer里面也有。不过这里用的是自己想出来的方法。
2、解法
这里涉及到快速查找肯定是想着使用二分查找。类比一维数组的二分查找,需要两个指针表示范围(也就是一条线段)。但是这里是二维数组,所以想着使用四个指针表示范围(矩阵有四个角)。另外,二分查找中肯定需要个中点,这个中心点自然而然选择矩形的中心点。在一维数组,中心点把数组分成了两个部分。这里在二维数组中,可以看作中心点把矩形分成了四个小矩形。那么如何进行二分查找呢?
这里如果中心点等于target,则很简单。如果中心点对应的值小于target,那么只能去除一个小矩形(如图左上角)。这里就发现没法像二分查找时候使用直接迭代的方法了。所以这里使用递归调用,在剩下的三个小矩形中进行递归查找。如果中心点对应的值大于target,同理分析即可。
另外有一个可以优化的地方。容易发现后面三个递归可以变成另个递归调用。因为下面的两个小矩形可以组成一个大矩形(或者右侧的两个小矩形也可以组成一个大矩形)。
一步步思路,很自然。具体见代码。
class Solution {
public:
bool fun(vector<vector<int>>& matrix, int target,int r1,int c1,int r2,int c2){
int r_m,c_m;
while(r1<=r2&&c1<=c2){
r_m=r1+(r2-r1)/2;
c_m=c1+(c2-c1)/2;
if(matrix[r_m][c_m]==target){
return true;
}else if(matrix[r_m][c_m]>target){
return fun(matrix,target,r1,c1,r2,c_m-1)||fun(matrix,target,r1,c_m,r_m-1,c2);
}else{
return fun(matrix,target,r_m+1,c1,r2,c2)||fun(matrix,target,r1,c_m+1,r_m,c2);
}
}
return false;
}
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int r1=0,c1=0,r2=matrix.size()-1,c2=matrix[0].size()-1;
return fun(matrix,target,r1,c1,r2,c2);
}
};