AcWing 789.数的范围

本题为y总算法基础课对于整数二分的练习题
原题链接如下：
AcWing 789. 数的范围
题目如下：

给定一个按照升序排列的长度为 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,m;
int q[N];
//主函数
int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)scanf("%d",&q[i]);
    while(m--)
    {
        int x;//这是输入的需要求起始和终止位置的数字
        scanf("%d",&x);
        //先求起始位置，也就是小于等于这个数的区间的右边界
        int l=0,r = n-1;
        while(l<r)
        {
            int mid = l+r>>1;
            if(q[mid]>=x) r= mid;
            else l = mid+1;
        }
        //如果你要找的数 压根就不在这个数组里面，则二分到最后二分区间为1得出来的那个数也不是正确答案
        if(q[l]!=x) cout<<"-1 -1"<<endl;
        else
        {
           //先输出一下上面求出来的起始位置
           cout<<l<<" ";
           //现在开始求终止位置
           int l = 0;r = n-1;
           while(l<r)
           {
               //注意 这里对mid的取值涉及到边界问题 我们先用l+r，之后再根据两个元素的特殊情况来判断要不要向上取整
               int mid = l+r+1>>1;
               if(q[mid]<=x) l = mid;//这里是l=mid，我们利用两个元素的边界情况，来判断此时mid初始取值必须l+r+1>>1
               else r = mid - 1;
           }
           cout<<l<<endl;
           
        }
    }
    return 0;
}

最后这里将一下边界情况的处理，我们会发现求两个位置的时候，mid变量的取法一个是l+r，一个是l+r+1，为什么会出现l+r+1这种我们以前不太熟悉的情况呢，我们可以用只含有两个元素的边界情况来说明，比如1和2，1对应的下标为0，2对应的下标为1，则此时l+r和l+r+1分别为0和1，如果我们x取1，同时取mid = l+r的话，q[mid]此时就是1，然后符合if语句里面的条件，则l的值更新为mid，也就是0又更新成了0，陷入了死循环，将永远运行得不出答案，直至超时。
由于本人表达能力有限，具体对边界情况的解释，可参考y总视频
链接如下：y总二分讲解：二分在视频最后一部分