题目:
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1
找第一次出现k值的位置:
int l=0,r=n-1;
while(l<r)
{
int mid=l+r>>1;
if(a[mid]>=k)r=mid;
else l=mid+1;
}
找最后一次出现k值的位置:
l=0,r=n-1;
while(l<r)
{
int mid=l+r+1>>1;
if(a[mid]<=k)l=mid;
else r=mid-1;
}
注意这里的mid=l+r+1>>1;
为什么要加一呢?
因为l+r>>1是向上取整,若出现:
例如:
此情况 l一直为9跳不出循环,所以需要+1使它变为向下取整就可以避免此情况的发生;
代码:
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N];
int main()
{
int n,q;
cin>>n>>q;
for(int i=0;i<n;i++)cin>>a[i];
while(q--)
{
int k;
cin>>k;
int l=0,r=n-1;
while(l<r)
{
int mid=l+r>>1;
if(a[mid]>=k)r=mid;
else l=mid+1;
}
if(a[l]!=k)
{
cout<<"-1 -1"<<endl;
continue;
}
cout<<l<<' ';
l=0,r=n-1;
while(l<r){
int mid=l+r+1>>1;
if(a[mid]<=k)l=mid;
else r=mid-1;
}
cout<<l<<endl;
}
return 0;
}