洛谷P1102——STL中两个二分查找函数

1.介绍

 假定我们有一个有序的数组a，并将数x作为二分查找的目标，那么：

（1）lower_bound

lower_bound(a,a+n,x)-a      //下标从0开始
lower_bound(a+1,a+n+1,x)-a  //下标从1开始

它们就能取得最小的数组a的下标i，满足ai​⩾x。

即找到第一个大于等于x的数组下标。

（2）upper_bound

upper_bound(a,a+n,x)-a      //下标从0开始
upper_bound(a+1,a+n+1,x)-a  //下标从1开始

它们就能取得最小的数组a的下标i，满足ai​>x。

即找到第一个大于x的数组下标。

#易错提醒

lower_bound(a+1,a+n+1,x)-a    函数返回值是指针类型，减去数组名（头指针）得到的才是查找结果的下标。

2.代码

#include<iostream>
#include<algorithm>
using namespace std;

int n,c;
long long ans,a[200010];

int main()
{
	int i;
	cin>>n>>c;
	for (i=1;i<=n;i++)
		cin>>a[i];
	sort(a+1,a+n+1);
	for (i=1;i<=n;i++)
		ans+=upper_bound(a+1,a+n+1,a[i]+c)-lower_bound(a+1,a+n+1,a[i]+c);
	cout<<ans;
	return 0;
}