编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,50]], target = 3
输出:true
思路:
采用二分法,难点在于如何将二维数组转化为一位。m x n 矩阵一维索引 转 二维坐标(row, col) row = index / n, col = index % n。剩下就是二分法while(left<right)。
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.empty()||matrix[0].empty()) return false;
int m=matrix.size(),n=matrix[0].size();
int left=0;
int right=m*n-1;
while(left<right){
int mid=left+right>>1;
if(matrix[mid/n][mid%n]>=target){
right=mid;
}
else{
left=mid+1;
}
}
if(matrix[left/n][left%n]==target) return true;
return false;
}
};