如果对于一段序列,我们可以按照某种性质将其分成【具备该性质】和【不具备该性质】的两段子序列,而二分法可以用于寻找这两段子序列的边界。
下面给出两段代码
//用于寻找具备某性质的子序列的下边界
int bsearch_lb(int l, int r)//lowerbound
{
while (l < r)
{
int mid = l + r >> 1;
if (check(mid)) r = mid; // check()判断mid是否满足性质
else l = mid + 1;
}
return l;
}
//用于寻找具备某性质的子序列的上边界
int bsearch_ub(int l, int r)//upper bound
{
while (l < r)
{
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
假设一段序列的排列是:【具备某性质】【不具备某性质】,我们要找的边界就是具备某性质的子序列的上边界,因此使用bsearch_ub的函数,其中当mid满足某性质时,check()函数返回1,否则返回0.
如果一段序列是:【不具备某性质】【具备某性质】,那么我们要找的边界时具备某性质的子序列的下边界,使用bsearch_lb的函数。
例题:
给定一个按照升序排列的长度为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
#include <iostream>
using namespace std;
const int N = 100000;
int num[N];
int bsearch_lb(int num[],int l,int r,int tar)//将数组分割成【小于tar】【大于等于tar】两部分,我们要找到【大于等于tar】的下界
{
while(l<r)
{
int mid = l+r>>1;
if(num[mid]>=tar)r=mid;//寻找下界,如果满足条件就继续往左缩小范围,寻找下界
else l=mid+1;
}
return num[l]==tar?l:-1;
}
int bsearch_ub(int num[],int l,int r,int tar)//将数组分割成【小于等于tar】【大于tar】两部分,我们要找到【小于等于tar】的上界
{
while(l<r)
{
int mid=l+r+1>>1;
if (num[mid]<=tar)l=mid;//如果满足条件就继续往右缩小范围,寻找上界
else r=mid-1;
}
return num[l]==tar?l:-1;
}
int main()
{
int n,q;
cin>>n>>q;
for(int i=0;i<n;i++)cin>>num[i];
for(int i=0;i<q;i++)
{
int ques;
cin>>ques;
int lb,ub;
lb = bsearch_lb(num,0,n-1,ques);
ub = bsearch_ub(num,0,n-1,ques);
cout<<lb<<' '<<ub<<endl;
}
return 0;
}