题目链接:http://codeforces.com/group/NVaJtLaLjS/contest/238855/problem/A
题目大意:给你三个数组a,b,c,以及一个d,求有多少个三元组(a[i],b[j],c[k])满足|a[i]-b[j]| <= d,|a[i]-c[k]| <= d,|b[j]-c[k]| <= d;
题面:
题目分析:可以先假设a[i]为三个数中最小的,则方案数为:设b[j]中大于等于a[i]并且小于等于a[i]+d的数的个数记为cnt1, c[k]中大于等于a[i]并且小于等于a[i]+d的数的个数记为cnt2,ans += cnt1*cnt2;分别枚举a[i]为最小或者b[j]为最小或者c[k]为最小,注意去掉重复的即可。
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e5+7;
typedef long long ll;
int a[maxn],b[maxn],c[maxn];
int main(){
int d,A,B,C;
while(scanf("%d%d%d%d",&d,&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]+d)-lower_bound(b,b+B,a[i]);
ll cnt2 = upper_bound(c,c+C,a[i]+d)-lower_bound(c,c+C,a[i]);
//cout<<cnt1*cnt2<<endl;
ans += cnt1*cnt2;
}
for(int i = 0;i < B; i++){
ll cnt1 = upper_bound(a,a+A,b[i]+d)-upper_bound(a,a+A,b[i]);
ll cnt2 = upper_bound(c,c+C,b[i]+d)-lower_bound(c,c+C,b[i]);
//cout<<cnt1*cnt2<<endl;
ans += cnt1*cnt2;
}
for(int i = 0;i < C; i++){
ll cnt1 = upper_bound(b,b+B,c[i]+d)-upper_bound(b,b+B,c[i]);
ll cnt2 = upper_bound(a,a+A,c[i]+d)-upper_bound(a,a+A,c[i]);
//cout<<cnt1*cnt2<<endl;
ans += cnt1*cnt2;
}
cout<<ans<<endl;
}
return 0;
}