有趣的遊戲

【簡要題面】有四組數字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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章