LeetCode 74. 搜索二维矩阵(双二分查找二维数组)

两次二分查找
由题目已知矩阵每行中整数升序排列,行之间也升序排列,得解题思想:
先对第一列进行一次二分查找,确定目标元素所在行;然后对该行二分查找,确定是否存在目标元素。

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        if(target < matrix[0][0]) return false;  //目标值小于第一个元素,必不存在,直接返回
        int m = matrix.size();      //m*n的二维矩阵
        int n = matrix[0].size();
        int i,j;                    //目标元素的横纵坐标

        int l = 0,r = m-1;          //第一次二分查找,确定所在行
        while(r-l > 1) {            //由于二分后半部分不跳中间行的特殊处理,循环最终可能l、r保持相距一位而不交集,故l、r相距不到2位时就退出
            i = (l+r)/2;
            if(matrix[i][0] > target) r = i-1;
            else if(matrix[i][0] < target) l = i;  //因为target必大于等于所在行第一个元素,二分截取后半部分时不跳过中间行
            else return true;                      //该行第一个元素正好是目标元素,直接返回
        }
        if(target >= matrix[r][0]) i = r;  //目标元素若存在,一定在l行或r行,令i=合适的行
        else i = l;

        l = 0;           //第二次二分查找,找到目标元素
        r = n-1;
        while(l <= r) {  //普通的顺序表二分查找
            j = (l+r)/2;
            if(matrix[i][j] > target) r = j-1;
            else if(matrix[i][j] < target) l = j+1;
            else return true;  //找到了返回
        }
        return false;  //未找到返回
    }
};

时间复杂度O(log m + log n = log(mn)),空间复杂度O(1)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值