题目描述:
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true
示例 2:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
输出: false
思路:采取两次二分法,第一次先查找属于哪个vector,即组间进行二分查找;第二次再在组内进行二分查找
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
//先组间进行二分查找,再组内进行二分查找
int row=matrix.size();
if(row==0)
return false;
int col=matrix[0].size();
if(col==0)
return false;
int low=0,high=row-1;
while(low<=high){
int mid=(low+high)/2;
if(matrix[mid][0]<=target)
low=mid+1;
else
high=mid-1;
}
if(high<0)//需要注意如果比第一行的第一个元素都小,则查找失败
return false;
int left=0,right=col-1;
while(left<=right){
int mid=(left+right)/2;
if(matrix[high][mid]==target)
return true;
else if(matrix[high][mid]<=target)
left=mid+1;
else
right=mid-1;
}
return false;
}
};