[C++] LeetCode 74. 搜索二维矩阵

题目

编写一个高效的算法来判断 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

思路解析

这题主要考察二分法，先用二分法确定target所在的行，然后在所在的行内用二分法确定是否存在

代码

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 up=0,bottom=m,mid;
        while(up<bottom){
            mid=up+(bottom-up)/2;
            if(matrix[mid][0]==target) return true;
            else if(matrix[mid][0]<target) up=mid+1;
            else bottom=mid;
        }
        mid=up-1;
        if(mid<0) return false;
        auto &vec=matrix[mid];
        auto it=lower_bound(vec.begin(),vec.end(),target);
        if(it==vec.end()||*it!=target) return false;
        else return true;
    }
};