整数二分模板题----数的范围

Posted 2022-04-29 4nc414g0n

数的范围

题目：数的范围
描述：给定一个按照升序排列的长度为 n 的整数数组，以及 q 个查询。对于每个查询，返回一个元素 k 的起始位置和终止位置（位置从 0 开始计数）。如果数组中不存在该元素，则返回 -1 -1。

---

思路：

• 典型的二分模板题，
  先找左边第一个目标数字（[l,mid],[mid+1,r]，mid不用+1），
  再找右边第一个目标数字（[l,mid-1],[mid,r]，mid需要+1）

---

代码如下：

#include <iostream> 
using namespace std;
const int N = 1e5+10;//一般多开10个空间 int a[N];

void bsearch(int l,int r, int num,int size)
 
    while(l<r)//左边起第一个
    
        int mid= (l+r)>>1;
        if(a[mid]>=num) r=mid;
        else l=mid+1;//mid肯定不满足
    
    if(a[r] == num) 
    
        cout<<r<<" ";
        l=0;r=size-1;
        while(l<r)//右起第一个
        
            int mid=(1+l+r)>>1;
            if(a[mid]<=num) l=mid;
            else r=mid-1;//mid肯定不满足
        
        cout<<r<<endl;
    
    else 
        cout<<"-1 -1"<<endl;  int main() 
    int n=0;
    int q=0;
    cin>>n;
    cin>>q;
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    while(q--)
    
        int num=0;
        cin>>num;
        bsearch(0,n-1,num,n);
    
    return 0; 
    

创作打卡挑战赛 赢取流量/现金/CSDN周边激励大奖