2019 蓝桥杯省赛 B 组模拟赛 程序设计：好友

题目：


代码如下：

#include<bits/stdc++.h>
using namespace std;
int a[100005];
int main()
{
	long long n,d,sum = 0;
	cin >> n >> d;
	for(int i = 0;i < n;i++) cin >> a[i];
	sort(a,a + n);
	for(int i = 0;i < n;i++) sum += (upper_bound(a,a + n,a[i] + d) - a) - i - 1;
	cout << sum << endl;
	return 0;
}

这道题如果你用暴力每次回溯去计算次数，就算加一个优化，还是不能通过全部数据，因为暴力的复杂度达到了o(n^2)。这道题要用到二分的思想，首先将所有渔民房子的坐标从小到大排序，然后从第一个开始寻找比自己位置 + 活动半径d大的第一个数据的位置。
这里就要用到upper_bound函数了，先来简单介绍一下,upper_bound(a,a + n,x)代表的一下从数组a[0]到a[n]中寻找第一个>x的数的位置(这里数组a一定要先排序)，然后返回这个位置的地址。
那么这里找到比自己位置 + 活动半径d大的第一个数据的位置之后减去数组初始地址(也就是a)再减去i - 1，那就是和第i个渔民是好友的个数了。从第一个渔民一直到最后一个，每次都把数目累计，那就是最终答案了。总体复杂度是o(nlogn)。