力扣HOT100之二分查找:74. 搜索二维矩阵

力扣HOT100之二分查找:74. 搜索二维矩阵_第1张图片
这道题直接a了,我们可以参考上一道题:35.搜索插入位置的思路,详情见我的上一篇博客。将每一行的第一个元素当作一个数组中的元素,然后对这个数组进行二分查找,如果直接找到了target,则直接返回true,否则退出循环的时候left指向应当插入的下标位置,举几个例子:
1.当输入为

[[1,3,5,7],
 [10,11,16,20],
 [23,30,34,60]]

target3,则退出循环时left指向元素10所在的位置,此时我们应当去上一层(matrix[left - 1])寻找target
2.当输入为

[[1,3,5,7],
 [10,11,16,20],
 [23,30,34,60]]

target为61,则退出循环时left的值为3,此时我们应当去上一层(matrix[left - 1],即matrix[2])寻找target
3.当输入为

[[1,3,5,7],
 [10,11,16,20],
 [23,30,34,60]]

target为0,则退出循环时left的值为0,此时我们无法去上一层(matrix[-1],会发生溢出)寻找target。所以我们需要判断一下退出二分查找后的left是否大于0,只有大于0才进行进一步的搜索,当遍历完整层都没有找到时,返回false,若找到则直接返回true

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int left = 0, right = matrix.size();
        int mid;
        while(left < right){
            mid = (left + right) / 2;
            if(matrix[mid][0] > target)
                right = mid;
            else if(matrix[mid][0] < target)
                left = mid + 1;
            else return true;            
        }
        int i = 0;
        while(left > 0 && i < matrix[0].size()){
            if(matrix[left - 1][i] == target)
                return true;
            i++;
        }
        return false;
    }
};

你可能感兴趣的:(力扣HOT100,leetcode,矩阵,算法)