题目大意:
给你三个数组a,b,c(非降序排列),分别包含na, nb, nc个整数。问你能找到多少个这样的三元组(i, j, k),使得ai, bj, ck三者两两之间绝对值之差不超过d。
解题思路:
显然ai, bj, ck这三个数必然在区间[i, i+d]内。有三种计数情况:1:ai <= bj <= ck ; 2 : ai > bj 且 ck >= bj;3: ai > ck 且 bj > ck。(可以理解为排列的开头不同)根据以上三种情况进行计数。要迅速找出一个区间内有多少个数,二分即可。只需要注意lower_bound和upper_bound分别就好 。
代码如下:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[500005], b[500005], c[500005];
int n, A, B, C;
int main(){
while(scanf("%d %d %d %d", &n, &A, &B, &C) != EOF){
for(int i=0;i<A;++i) scanf("%d", &a[i]);
for(int i=0;i<B;++i) scanf("%d", &b[i]);
for(int i=0;i<C;++i) scanf("%d", &c[i]);
ll ans = 0;
for(int i=0;i<A;++i){
ll cnt1 = upper_bound(b, b+B, a[i]+n) - lower_bound(b, b+B, a[i]);
ll cnt2 = upper_bound(c, c+C, a[i]+n) - lower_bound(c, c+C, a[i]);
ans += cnt1*cnt2;
// cout << cnt1*cnt2 << endl;
}
for(int i=0;i<B;++i){
ll cnt1 = upper_bound(a, a+A, b[i]+n) - upper_bound(a, a+A, b[i]);
ll cnt2 = upper_bound(c, c+C, b[i]+n) - lower_bound(c, c+C, b[i]);
ans += cnt1*cnt2;
// cout << cnt1*cnt2 << endl;
}
for(int i=0;i<C;++i){
ll cnt1 = upper_bound(b, b+B, c[i]+n) - upper_bound(b, b+B, c[i]);
ll cnt2 = upper_bound(a, a+A, c[i]+n) - upper_bound(a, a+A, c[i]);
ans += cnt1*cnt2;
// cout << cnt1*cnt2 << endl;
}
cout << ans << endl;
}
return 0;
}