欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

有趣的游戏

程序员文章站 2022-04-19 15:36:50
...

【简要题面】有四组数字abcda,b,c,da,b,ca,b,c的个数小于1000,d的个数小于1e5。对于每个d求是否存在ai+bi+ci=da_i+b_i+c_i=daibicia_i、b_i、c_i均为abca、b、c数组中的一个数)。

【分析】
由于abca、b、c中数字的大小未定,所以我们很可惜的不能用f数组进行预处理,实际上复杂度也不允许我们这么做。比较巧妙的做法是先将aba、b中的数字先相加用map存,再在读入d数组的时候枚举a数组和d数组相减,并在map中查询是否存在diajd_i-a_j。鉴于map自带logn+stllog_n+stl常数所以我们可以用HashtableHash table代为实现。

【code】

#pragma GCC optimize(2)
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1100;
const int mod=1e6+7;
const int key=1e6+9;
int la,lb,lc,ld;
int a[maxn],b[maxn],c[maxn];
struct hash_table{
	struct node{
		int val,nxt;
	}num[mod<<1];
	int head[mod],tot;
	void add(int x){int p=1ll*x*key%mod;num[++tot]=(node){x,head[p]},head[p]=tot;}
	bool find(int x){
		int p=1ll*x*key%mod;
		for(int i=head[p];i;i=num[i].nxt){
			if(num[i].val==x)return 1;
		}
		return 0;
	}
}mp;
inline void read(int &x){
	x=0;char tmp=getchar();int fl=1;
	while(tmp<'0'||tmp>'9'){if(tmp=='-')fl=-fl;tmp=getchar();}
	while(tmp>='0'&&tmp<='9') x=(x<<1)+(x<<3)+tmp-'0',tmp=getchar();
	x=x*fl;
}
inline bool check(int x){
	for(int i=1;i<=la;i++){
		if(x-a[i]<0) break;
		if(mp.find(x-a[i]))return 1;
	}
	return 0;
}
int main(){
//	freopen("game.in","r",stdin);
//	freopen("game.out","w",stdout);
	cin>>la>>lb>>lc;
	for(int i=1;i<=la;i++) read(a[i]);
	for(int i=1;i<=lb;i++) read(b[i]);
	for(int i=1;i<=lc;i++) read(c[i]);
	if(la>lb) swap(a,b),swap(la,lb);
	if(la>lc) swap(a,c),swap(la,lc);
	for(int i=1;i<=lb;i++)
		for(int j=1;j<=lc;j++)
			if(!mp.find(b[i]+c[j]))
			mp.add(b[i]+c[j]);
	sort(a+1,a+la+1);
	cin>>ld;
	while(ld--){
		int x;read(x);
		if(check(x)) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}