题目描述
给定一个按照升序排列的长度为n的整数数组,以及 q 个查询。
对于每个查询,返回一个元素k的起始位置和终止位置(位置从0开始计数)。
如果数组中不存在该元素,则返回“-1 -1”。
输入格式
第一行包含整数n和q,表示数组长度和询问个数。
第二行包含n个整数(均在1~10000范围内),表示完整数组。
接下来q行,每行包含一个整数k,表示一个询问元素。
输出格式
共q行,每行包含两个整数,表示所求元素的起始位置和终止位置。
如果数组中不存在该元素,则返回“-1 -1”。
数据范围
1≤n≤100000
1≤q≤10000
1≤k≤10000
样例
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1
注意本题的意思,让输出的是,某个数出现的起始位置和结束位置,即查找大于等于x的最小坐标(起始位置)和查找小于等于x的最大坐标(结束位置),若不存在没有出现x,则输出-1 -1。
最初因为理解错了题目的意思导致错误的思路,害,语文永远是硬伤呀^_^
模板:
查找大于等于x的最小坐标l r
int l=0,r=n-1;
while(l<r){ //查找大于等于x的最小坐标
int mid=(l+r)/2;
if(a[mid]>=x) r=mid;
else l=mid+1;
}
查找大于x的最小坐标 l r
int l=0,r=n-1;
while(l<r){ //查找大于x的最小坐标
int mid=(l+r)/2;
if(a[mid]>x) r=mid;
else l=mid+1;
}
查找小于等于x的最大坐标l r
int l=0;r=n-1;
while(l<r){ //查找小于等于x的最大坐标
mid=(l+r+1)/2;//注意加+1,因为/向下取整,若l=r-1,mid=(l+r)/2=l,满足条件时,l=mid,mid=(l+r)/2还是等于l,死循环
if(a[mid]<=x) l=mid;
else r=mid-1;
}
查找小于x的最大坐标l r
int l=0;r=n-1;
while(l<r){ //查找小于x的最大坐标
mid=(l+r+1)/2;//注意加+1,因为/向下取整,若l=r-1,mid=(l+r)/2=l,满足条件时,l=mid,mid=(l+r)/2还是等于l,死循环
if(a[mid]<x) l=mid;
else r=mid-1;
}
C++ 代码
#include <iostream>
#include <cstdio>
using namespace std;
int m,n;
int x;
int a[100005];
int main(){
cin>>n>>m;
for(int i=0;i<n;i++){
cin>>a[i];
}
while(m--){
cin>>x;
int l=0;
int r=n-1;
int mid;
while(l<r){ //查找大于等于x的最小坐标
mid=(l+r)/2;
if(a[mid]>=x) r=mid;
else l=mid+1;
}
if(a[l]!=x) cout<<-1<<" "<<-1<<endl;
else{ //l,r相等,写l或者r都行
cout<<l<<" ";
l=0;r=n-1;
while(l<r){ //查找小于等于x的最大坐标
mid=(l+r+1)/2;
if(a[mid]<=x) l=mid;
else r=mid-1;
}
cout<<l<<endl;
}
}
return 0;
}
参考: 超详细的二分题解