NYOJ - 找球号(二）(HASH)

描述

在某一国度里流行着一种游戏。游戏规则为：现有一堆球中，每个球上都有一个整数编号i(0<=i<=100000000),编号可重复，还有一个空箱子，现在有两种动作：一种是"ADD"，表示向空箱子里放m(0<m<=100)个球，另一种是"QUERY”，表示说出M(0<M<=100)个随机整数ki(0<=ki<=100000100),分别判断编号为ki 的球是否在这个空箱子中（存在为"YES",否则为"NO"），先答出者为胜。现在有一个人想玩玩这个游戏，但他又很懒。他希望你能帮助他取得胜利。

输入

第一行有一个整数n(0<n<=10000);
随后有n行;
每行可能出现如下的任意一种形式：
第一种：
一个字符串"ADD"，接着是一个整数m，随后有m个i；
第二种：
一个字符串"QUERY”，接着是一个整数M，随后有M个ki；

输出

输出每次询问的结果"YES"或"NO".

样例输入

2
ADD 5 34 343 54 6 2
QUERY 4 34 54 33 66

样例输出

YES
YES
NO
NO

#include <stdio.h>
#include <string.h>#define N 1000001
#define MOD 200003
int Hash[N],Head[N],Next[N];
int top;void add(int num)
{int key = num % MOD;Next[top] = Head[key];Head[key] = top;Hash[top] = num;top++;
}int query(int num)
{int key = num % MOD;for(int i = Head[key]; i >= 0; i = Next[i]){if(Hash[i] == num)return 1;}return 0;
}
int main()
{int ncase;char s[10];scanf("%d",&ncase);memset(Head,-1,sizeof(Head));while(ncase--){int n,m;scanf("%s%d",s,&n);if(s[0] == 'A'){for(int i = 0; i < n; i++){scanf("%d",&m);add(m);}}else{for(int i = 0; i < n; i++){scanf("%d",&m);printf("%s\n",query(m)?"YES":"NO");}}}return 0;
}