P1296 奶牛的耳语 （快排+二分查找）

题目链接：https://www.luogu.com.cn/problem/P1296

解法一：

朴素的想法，枚举每一头牛，假设位置为 x ，可以与这头牛交流的牛的数量为：在区间 [x - d, x + d] 内的牛。于是问题转变为，如何快速求出给定区间内有多少头牛。

先将所有牛排个序，然后二分查找，找到大于等于 x-d 最左边的牛编号，小于等于 x+d的最右边的牛的编号，即可得出区间内有多少头牛。

时间复杂度：nlogn  空间复杂度：o(n)

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<set>
#include<cstring>
using namespace std;

const int maxn = 1e6; 
int a[maxn + 10], n, d;

int binaryleft(int x) {
	int l = 0, r = n - 1, mid;
	while(l <= r) {
		int mid = (l + r) / 2;
		if(a[mid] < x) l = mid + 1;
		else r = mid - 1;
	}
	return l;
}

int binaryright(int x) {
	int l = 0, r = n - 1, mid;
	while(l <= r) {
		int mid = (l + r) / 2;
		if(a[mid] > x) r = mid - 1;
		else l = mid + 1;
	}
	return r;
}

int main() {
	scanf("%d%d", &n, &d);
	for(int i = 0; i < n; i++) scanf("%d", &a[i]);
	sort(a, a + n);
	
	int ans = 0;
	for(int i = 0; i < n; i++) {
		int l = binaryleft(a[i] - d);
		int r = binaryright(a[i] + d);
		ans += r - l;
	}
	printf("%d\n", ans / 2);
	return 0;
}

解法二：

我们维护一个窗口，代表第 i 头牛向右可以沟通到的牛，假设第 i 头牛向右最远可以沟通到第 j 头牛。

那么第 i+1 头牛至少可以向右沟通到第 j 头牛，我们直接从 j 开始向后枚举，得到第 i+1 头牛向右最远可以沟通到的牛。

时间复杂度：o(nlogn)   空间复杂度:o(n)

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<set>
#include<cstring>
using namespace std;

const int maxn = 1e6; 
int a[maxn + 10], n, d;

int main() {
	scanf("%d%d", &n, &d);
	for(int i = 0; i < n; i++) scanf("%d", &a[i]);
	sort(a, a + n);
	
	int ans = 0, r = 0;
	for(int i = 0; i < n; i++) {
		for(;r < n && a[i] + d >= a[r]; r++);
		ans += r - i - 1;
	}
	printf("%d\n", ans);
	return 0;
}