30天学会基础算法 (A-B 数对—双指针&集合) ^10^

题目描述

给出一串数以及一个数字 C，要求计算出所有 A - B = C 的数对的个数（不同位置的数字一样的数对算不同的数对）。
对于 75% 的数据，1≤N≤2000。
对于 100% 的数据，1≤N≤2×10⁵ 。
保证所有输入数据绝对值小于 2³⁰ ，且 C≥1。

输入格式

输入共两行。
第一行，两个整数 N, C。
第二行，N 个整数，作为要求处理的那串数。

输出格式

一行，表示该串数中包含的满足 A - B = C 的数对的个数。

输入样例

4 1
1 1 2 3

输出样例

3

题目分析

我们考虑题目要求求出所有A-B=C的数对，我们可以先将原数组排序，然后就会发现每个数A，对应的数B一定是一段连续的区间。

然后我们再考虑如何去找到这个区间。

我们显然是要找到这个连续区间的左端点和右端点。

考虑到排序之后序列的有序性，我们枚举每个数，他们的左端点和右端点都是单调不降的，因此我们可以用双指针来维护这个东西。

具体的实现就是，我们维护两个右端点r1 , r2，每次r1右移到a[r1] - a[l] <= c的最后位置的下一位，r2右移到满足a[r2] - a[l] < c最后一位.

也就是说， 此时如果a[r2] - a[l] == c && a[r1 - 1] - a[l] == c，中间的那一段一定都是满足条件的，我们让ans += r1 - r2即可。

code

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;
int n, c;
int a[N];

int main()
{
	cin >> n >> c;
	for (int i = 1; i <= n; i++) cin >> a[i];
	sort(a + 1, a + 1 + n);
	int l = 1, r1 = 1, r2 = 1;
	ll ans = 0;
	for (l = 1; l <= n; l++) 
	{
		while (r1 <= n && a[r1] - a[l] <= c) r1++;
		while (r2 <= n && a[r2] - a[l] < c) r2++;
		if (a[r2] - a[l] == c && a[r1 - 1] - a[l] == c && r1 - 1 >= 1)
			ans += r1 - r2;
	}
	cout << ans << endl;
	return 0;
}