简短优美的二分模板

二分查找

二分查找可以在满足单调性的数组中查找给定的值,这里的单调性不一定是指数值大小,也可以指某些性质,比如前面的值满足某个性质,那么后面的值也满足该性质,那么就可以称其具有单调性。

时间复杂度为O(logn)

下面分享一个简洁易记的二分查找模板,以一个升序的数组为列,有以下两种查找方式。
 

  1. 查找第一个小于等于k的数值的下标。
    在这里插入图片描述
    如上图所示蓝色区域为小于k的,橙色区域为等于k的,红色区域为大于k的。
int l = 0, r = n-1;
while (l < r)
{
    int mid = (l + r) >> 1;
    if (nums[mid] >= k) r = mid;
    else l = mid + 1;
}
  1. 查找最后一个小于等于k的数值的下标
    在这里插入图片描述
int l = 0, r = n-1;
while (l < r)
{
  	int mid = (l + r + 1) >> 1;
  	if (nums[mid] <= k) l = mid;
 	else r = mid - 1;
}

注意,第二种方法中需将(l+r)/2进行上取整,否则当l与r相邻且nums[mid]<=k时,mid仍为l,使得l,r值都不改变而继续重复之前过程,导致死循环。
 

下面附上一道例题

题解:

#include <iostream>
using namespace std;

const int N = 100010;
int nums[N];
int n, q, k;


int main ()
{
    cin >> n >> q;
    for (int i = 0; i < n; i++) cin >> nums[i];
    
    for (int i = 0; i < q; i++)
    {
        cin >> k;

        // 查找起始位置
        int l = 0, r = n-1;
        while (l < r)
        {
            int mid = (l + r) >> 1;
            if (nums[mid] >= k) r = mid;
            else l = mid + 1;
        }
        l = nums[l] == k? l : -1;
        cout << l  << " ";
        
        // 查找终止位置
        l = 0, r = n-1;
        while (l < r)
        {
            int mid = (l + r + 1) >> 1;
            if (nums[mid] <= k) l = mid;
            else r = mid - 1;
        }
        l = nums[l] == k? l : -1;
        cout << l << endl;
    }
    
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值