C++基本算法之二分

二分——有单调性的一定可以二分

例题：数的范围

题目描述
给定一个按照升序排列的长度为 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

解题过程

• l r 为左右边界 check（mid）
• mid =(l+r)/2 r=mid l=mid+1
• mid=(l+r+1)/2 l=mid r=mid-1

/*本题在数组中查找某元素，找不到就输出−1−1，
找到了就输出不小于该元素的最小位置和不大于该元素的最大位置。
所以，需要写两个二分，一个需要找到>=x>的第一个数，
另一个需要找到<=x的最后一个数。*/
#include<iostream>
using namespace std;
const int N=100010;
int q[N];
int n,m;
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
    cin>>q[i];
    while(m--)
    {
        int x;
        cin>>x;
        int l=0,r=n-1;
        while(l<r)//第一个二分求的是大于等于x的第一个位置
        {
            int mid=l+r>>1;
            if(q[mid]>=x)
            r=mid;
            else l=mid+1;
        }
        if(q[l]!=x)
        cout<<"-1 -1"<<endl;
        else//第二个二分求的是小于等于x的最后一个位置
        {
            cout<<l<<' ';
            int l=0,r=n-1;
            while(l<r)
            {
                int mid=l+r+1>>1;
                if(q[mid]<=x)
                    l=mid;
                    else
                    r=mid-1;
            }
        cout<<l<<endl;
    }
    }
    return 0;
}