题目
给定一个按照升序排列的长度为n的整数数组,以及 q 个查询。对于每个查询,返回一个元素k的起始位置和终止位置(位置从0开始计数)。如果数组中不存在该元素,则返回“-1 -1”。
输入格式
第一行包含整数n和q,表示数组长度和询问个数。
第二行包含n个整数(均在1~10000范围内),表示完整数组。
接下来q行,每行包含一个整数k,表示一个询问元素。
输出格式
共q行,每行包含两个整数,表示所求元素的起始位置和终止位置。
如果数组中不存在该元素,则返回“-1 -1”。
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1
注意:
- 二分法确定左边界时, r 和 l 的赋值
- 二分法确定右边界时, center = ( r + l + 1)/ 2,因为不加一,l 和 center值一直不变,进入死循环
#include <iostream>
using namespace std;
int main()
{
int n, q;
cin >> n >> q;
int A[100100];
for (int i = 0 ; i < n; i++) cin >> A[i];
while (q--) {
int a;
cin >> a;
int l = 0, r = n - 1;
if (A[0] == a || A[n - 1] == a) {
if (A[0] == a) {
int i;
for (i = 0; i < n; i++) if (A[i] != a) break;
cout << 0 << " " << i - 1 << endl;
}else {
int i;
for (i= n - 1; i >= 0; i--) if (A[i] != a) break;
cout << i + 1 << " " << n - 1 << endl;
}
}else {
while (l < r) {
int center = (l + r) / 2;
if (A[center] >= a) r = center;
else l = center + 1;
}
if (A[l] != a) cout << -1 << " " << -1 << endl;
else {
cout << l << " " ;
l = 0, r = n - 1;
while (l < r) {
int center = (l + r + 1) / 2;
if (A[center] <= a) l = center;
else r = center - 1;
}
cout << r << endl;
}
}
}
return 0;
}