STL使用方法
算法类
二分查找 lower_bound
和upper_bound
lower_bound
在#include<cstdio>
和using namespace std;
中
upper_bound
在#include<algorithm>
中
对单调线性数据进行查找 (默认数据从小到大)
数组:
lower_bound(区间开始位置,区间开始结束位置+1,key)
$[begin,end)$
lower_bound
返回最后一个不比key元素小的元素的位置(迭代器)。也就是说,区间里有key元素就返回第一个key元素的位置,如果没有key元素就返回小于key元素的最后一个最大元素的位置。
upper_bound(区间开始位置,区间开始结束位置+1,key)
$[begin,end)$
upper_bound
返回第一个比key元素大的元素的位置(迭代器)。也就是说,不管区间有没有key元素都会返回第一个比key元素大的元素的位置。
STL容器
lower_bound(名称.begin(),名称.end(),key)
upper_bound(名称.begin(),名称.end(),key)
因为是迭代器,所以想要输出数据时要加*
。
如果想要输出在区间里的位置(第几个)upper_bound(名称.begin(),名称.end(),key)-名称
。
如果区间里没有满足条件的元素,就返回最后一个数据后面的位置。(在STL容器中就是end
)
时间复杂度:$log_2(n)$
#include<algorithm>
#include<cstdio>
using namespace std;
const int N=1e5+100;
int n,m;
int a[N];
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d",a+i);
sort(a+1,a+n+1);
for (int i=1;i<=m;i++)
{
int x;
scanf("%d",&x);
printf("lower_bound返回元素:");
printf("%d\n",*lower_bound(a+1,a+n+1,x));
printf("lower_bound返回元素在序列中位置:");
printf("%d\n",lower_bound(a+1,a+n+1,x)-a);
printf("upper_bound返回元素:");
printf("%d\n",*upper_bound(a+1,a+n+1,x));
printf("upper_bound返回元素在序列中位置:");
printf("%d\n",upper_bound(a+1,a+n+1,x)-a);
}
return 0;
}
结果: