题目描述
给出一串数以及一个数字 C,要求计算出所有 A - B = C 的数对的个数(不同位置的数字一样的数对算不同的数对)。
对于 75% 的数据,1≤N≤2000。
对于 100% 的数据,1≤N≤2×105 。
保证所有输入数据绝对值小于 230 ,且 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;
}