题目描述:
编写一个高效的算法来判断 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
思路:
1、判断目标值在哪一行
2、进行二分查找
代码:
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.size()<1||matrix[0].size()<1)return 0;
int m=matrix.size(),n=matrix[0].size()-1;
int left,right,loc=-1;
for(int i=0;i<m;i++)
{
if(target>=matrix[i][0]&&target<=matrix[i][n])
{
if(target==matrix[i][0]||target==matrix[i][n])return true;
left=1;
right=n-1;
loc=i;
break;
}
}
if(loc==-1)return false;
int mid;
while(left<=right)
{
mid=(left+right)/2;
if(matrix[loc][mid]==target)return true;
else if(matrix[loc][mid]>target)right=mid-1;
else left=mid+1;
}
return false;
}
};