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