A. Three Arrays(思维+二分)

题目链接: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;
} 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值