二分模版题

Posted 2021-03-11 doubest

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

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

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

输入格式

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

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

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

输出格式

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

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

数据范围

1≤n≤1000001≤n≤100000
1≤q≤100001≤q≤10000
1≤k≤10000

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
int const N = 100010;
int nums[N];
int n,m;


int main(void){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++){
        scanf("%d",&nums[i]);
    }
    for(int T=0;T<m;T++){
        int x;
        scanf("%d",&x);
        int l=0,r=n-1;
        int mid;
        while(l<r){
            mid = l + r >> 1;
            if(nums[mid] >= x) r = mid;
            else l = mid + 1;
        }
        
        if(nums[r] == x){
            printf("%d ",r);
            r = n-1;
            while(l<r){
                mid = l + r + 1 >> 1;
                if(nums[mid] <= x) l = mid;
                else r = mid - 1;
            }
            printf("%d
",r);
        }else{
            printf("%d %d
",-1,-1);
        }
    }
    
    
    return 0;
}