力扣74-搜索二维矩阵-C++高效搜索

一、运行结果

 

二、题目

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：

每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。

示例 1：
输入：matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
输出：true
示例 2：
输入：matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
输出：false

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-a-2d-matrix

三、思路

由于矩阵每行都是升序排序的，且第一个整数大于前一行的最后一个整数，也就说是每行数字所包括的范围是不重叠的（最小值到最大值之前），所以可以先确定该元素属于哪一行所包含的范围，再在该行进行二分搜索即可。确定所在行，可以从上往下查找，每行只看最后一个元素，若最后一个元素比target小，则看下一行的最后一个元素，从上往下出现的第一个 >=target 的行即为target所在的行（如果存在），然后对该行进行二分查找。

四、代码

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int len1 = matrix.size();
        int len2 = matrix[0].size();
        int row = 0;
        while(row < len1 && matrix[row][len2-1] < target){
            row++;   //找到target所在的行（如果存在的话）
        }
        if(row == len1) return false;
        int left = 0, right = len2 -1, mid;
        while(left <= right){   //在确定的行中二分查找
            mid = left + (right - left) / 2;
            if(matrix[row][mid] == target) return true;
            else if(matrix[row][mid] > target){
                right = mid -1;
            }else{
                left = mid + 1;
            }
        }
        return false;
    }
};