二分查找模板

算法描述

输入 非严格递增的整型有序数组 A = 1, 2, 2, 2. 3, 3, 5, 5, 6；待查找元素 x = 2；

输出 x第一次出现的下标和x最后一次出现的下标。

模板代码

int binary_search(int num){ //返回>=num的第一个下标(l, r]
    int l = -1; int r = n;
    while(l + 1!= r){
        int mid = l + r >> 1;
        if(a[mid] < num) l = mid;
        else r = mid;
    }
    return r;
}

---

例题

相关题目练习链接acwing：acwing-789 acwing-790

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

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

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

输入格式

第一行包含整数 n和 m，表示数组长度和询问个数。

第二行包含 n 个整数（均在 1∼10000范围内），表示完整数组。

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

输出格式

共 m 行，每行包含两个整数，表示所求元素的起始位置和终止位置。

如果数组中不存在该元素，则返回 -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 = 1e5 +10;
int a[N];
int n;
//二分查找参考：https://blog.csdn.net/WJPnb1/article/details/126360962?spm=1001.2014.3001.5502
int binary_search(int num){ //返回>=num的第一个下标(l, r]
    int l = -1; int r = n;
    while(l + 1!= r){
        int mid = l + r >> 1;
        if(a[mid] < num) l = mid;
        else r = mid;
    }
    return r;
}
int main(){
    int m;
    cin >> n >> m;
    for(int i = 0; i < n; i ++) cin >> a[i];
    while(m --){
        int num;
        cin >> num;
        int x1 = binary_search(num);
        if(x1 == n || a[x1] != num){
            cout << "-1 -1" << endl;
        }
        else{
            int x2 = binary_search(num + 1); //因为都是数组是整型的，所以二分查找num+1返回的下标再减1就是num最后出现的下标
            cout << x1 << " " << x2 - 1 << endl;
        }        
     
    }
    return 0;
}

二分查找算法参考：https://blog.csdn.net/WJPnb1/article/details/126360962?spm=1001.2014.3001.5502