二分法求数的范围

给定一个按照升序排列的长度为 n 的整数数组,以及 q 个查询。

对于每个查询,返回一个元素 k 的起始位置和终止位置(位置从 0 开始计数)。

如果数组中不存在该元素,则返回 -1 -1。

输入格式
第一行包含整数 n 和 q,表示数组长度和询问个数。

第二行包含 n 个整数(均在 1∼10000 范围内),表示完整数组。

接下来 q 行,每行包含一个整数 k,表示一个询问元素。

输出格式
共 q 行,每行包含两个整数,表示所求元素的起始位置和终止位置。

如果数组中不存在该元素,则返回 -1 -1。

数据范围
1≤n≤100000
1≤q≤10000
1≤k≤10000
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1

上代码:

#include<iostream>
using namespace std;
const int N=100010;
int n,q,k,a[N];
int main()
{
    cin>>n>>q;
    for(int i=0;i<n;i++)
    cin>>a[i];
    while(q--)
    {
        cin>>k;
        int l=0,r=n-1;
        while(l<r)
        {
            int mid= l + r >> 1;
            if(a[mid]>=k) r=mid;
            else l=mid+1;
        }
        if (a[l]!=k) cout<<"-1 -1"<<endl;
        else{
        cout<<l<<" ";
        int l=0,r=n-1;
        while(l<r)
        {
            int mid= l+r+1 >> 1;
            if(a[mid]<=k) l=mid;
            else r=mid-1;
        }
        cout<<l<<endl;
        }
    }
    return 0;
}

二分法是一个比较绕的的算法,思路很简单但是写起来总是写不对,今天我们通过这一道题来简单了解一下二分法。

情况一:

首先,我们要意识到二分法并不只适用于升序降序,它实质上是一个寻找边界的算法,只不过相较于其他序列,升序降序的序列更容易设置一个判断标准,我们通过这个标准来判断此序列的中点在左右哪一边(左边不符合这个标准,右边符合这个标准),如果在右边,说明mid(中间数)符合这一标准,我们所要寻找的边界可能在mid左边,也可能就是mid,于是我们把查找范围更新l到mid,如果mid在左边,说明这个序列中间的这个数并不符合这一标准,我们所要寻找的边界一定在mid的右边,于是把范围变成从mid+1到r,如此不断更新直到l等于r,此时l与r都指向了我们所要找的边界。

情况二:

还有另一种情况是,如果我们设置的标准左边是满足这个标准的,右边是不满足这个标准的,但是这个标准并不是我们想要的结果,比如说这道题我们要查找数字在数列中的最后一个位置,设置的标准是mid是否小于等于这个数,由于这个数列是从小到大排序的,因此左边的数列一定是符合mid小于等于k的,而我们需要找的并不是这个条件,而是要找大于k的那个临界值,这样的情况又和上面讲的情况有所不同。情况一样,我们首先判断mid是否满足这个条件,如果满足这个条件说明我们要寻找的临界值可能是mid,也有可能在mid右边,于是l=mid,如果mid不满足这个条件,说明我们要找的临界值在mid左边,所以r=mid-1。

情况一中,mid=l+r/2,情况二中,mid=l+r+1/2,这是因为满足标准的情况下,情况一是使右边界变化,情况二是使左边界变化,如果刚开始的l=0,r=1,情况二如果也是mid=l+r/2,刚开始mid=0,满足条件的情况下l=mid=0,没有变化,会一直循环下去,所以我们要在这里加以区分。

还是老样子上图解吧!针对这道题。
在这里插入图片描述
此时mid=l+r>>1=2,此时的标准是a[mid]是否大于等于3,很明显不满足标准,于是l=mid+1=3
在这里插入图片描述
此时mid=4,a[mid]大于3,于是r=mid=4
在这里插入图片描述
此时mid=3,a[mid]等于3,于是r=mid=3,此时l=r退出循环,由于a[l]=3,于是我们输出此时的l,即3在这个数列中第一次出现的位置。

我们再判断3在这个数列中最后一次出现的位置,一夜回到解放前我们再来一遍哈
在这里插入图片描述

虽然位置还是那么个位置,但是条件已经不是那个条件了,a[mid]等于3,改变不再是r而是l,l=mid=3
在这里插入图片描述

a[mid]等于3,所以还是l变
在这里插入图片描述

此时a[mid]>3,于是r=mid-1=4,此时l=r,它们指向的位置就是3最后出现的位置。
在这里插入图片描述

我们得到了正确的结果。

注意,当l与r重叠时,l与r指向的一定是边界值,如果l与r指向的数不等于我们所要寻找的数,那一定就意味着数组中没有这个数,我们直接输出(-1,-1)就可以。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值