链接:这里
AC CODE
#include <cstdio>
#include <iostream>
#include <cmath>
#include <cstring>
#include <algorithm>
#define LL long long
using namespace std;
int n;
LL c;
LL a[200005];
int main(){
cin >> n >> c;
for(int i = 1; i <= n; ++i){
scanf("%lld", &a[i]);
}
sort(a, a + n + 1);
LL ans = 0;
LL cnt = 0;
for(int l = 1, i = 1, j = 1; l <= n; ++l){
while(i <= n && a[i] - a[l] <= c) ++i;
while(j <= n && a[j] - a[l] < c) ++j;
cnt += i - j;
}
cout << cnt << endl;
return 0;
}
解题报告
寻找A - B = C的数对数,利用双指针算法。我们先固定一个数B
,然后向前找A
,使A - B = C
,注意到,给的数据可能会有重复的数字,所以我们要找的是一个由相同数字组成的序列的左右端点。从第一个数开始,定义l
指向第一个数,然后定义指针i
,从l + 1
开始往后找,直到a[i] - a[l] > c
,这时,i
指向我们要找的序列的右端点,接着定义指针j
,与i
相似,去找序列的左端点,直到a[j] - a[l] >= c
,找到后计数器加上序列长度。那如果找不到符合条件的A怎么办?若都不符合条件,那么i
,j
都指向n + 1
,计数器 + 0