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