
题目就是给出数组,求一个数的左边界和右边界。就是分别用二分查找找到目标数的首位和末尾。
cpp
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n,q,k;
int l, r;
int a[N];
int main(){
cin >> n >> q;
for(int i = 0; i < n; i ++ ) cin >> a[i];
while(q -- ){
cin >> k;
l = -1; r = n;
while (l + 1 != r) {
int mid = (l + r) / 2;
if (a[mid] >= k) r = mid;
else l = mid;
} //找数的左边界,跳出循环的情况应该是l指向目标数的前一位,r指向目标数的最左位。
if (a[r] != k) cout << "-1 -1" << '\n';
else{
cout << r << ' '; //输出左边界
l = -1; r = n;
while (l + 1 != r) {
int mid = (l + r) / 2;
if (a[mid] <= k) l = mid;
else r = mid;
} //找数的右边界,跳出循环的情况应该是l指向目标数的最右位,r指向目标数的后一位。
cout << l <<'\n';
}
}
return 0;
}
二分算法最重要的是弄清楚,跳出循环时,左右指针指向的位置。