题目描述
给定一个按照升序排列的长度为 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
算法1
(c++STL)
int N = 10, a[] = {1, 1, 2, 4, 5, 5, 7, 7, 9, 9}, x = 5;
int i = lower_bound(a,a+N,x)-a,
j=upper_bound(a, a + N, x) - a;
// a[0] ~ a[i - 1] 为小于x的元素,
a[i] ~ a[j -1]为等于x的元素,
a[j]~a[N-1]为大于x的元素
binary_search:二分查找。binary_search(v.begin(), v.end(), value),其中 value 为需要查找的值。
C++ 代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e6+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 num;
cin>>num;
int i=lower_bound(a,a+n,num)-a;
int j=upper_bound(a,a+n,num)-a;
if(!binary_search(a,a+n,num))
cout<<"-1 -1"<<endl;
else cout<<i<<" "<<j-1<<endl;
}
}