nyoj138 找球号(二) hash哈希表 模板题

找球号(二）

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 5

描述

在某一国度里流行着一种游戏。游戏规则为：现有一堆球中，每个球上都有一个整数编号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<cstdio>
#include<cstring>

#define mod 200003
const int N=1e6+10;//100010 
int head[N],next[N],hash[N];
int n;//全局变量 


void add(int num)//核心代码  仔细理解 可以在纸上推导一下   2 3 8  mod=5 的情况。类似形成特征值数组，每个数组里下面一串链表，链着所有特征值相同的数字 
{
	int key=num%mod;//对mod取余得到关于num的特征值 
	hash[n]=num;
	next[n]=head[key]; 
	head[key]=n;
	n++;
}

int main()
{
	int T,m,a;
	char q[10];
	while(scanf("%d",&T)!=EOF)
	{
		memset(head,-1,sizeof(head));//初始化数组 
		n=0;//元素个数 
		while(T--)
		{
			scanf("%s %d",q,&m);
			if(q[0]=='A')//增加 
			{
				while(m--)
				{
					scanf("%d",&a);
					add(a);//处理 
				}
			}
			else//查询 
			{
				while(m--)
				{
					int flag=0;
					scanf("%d",&a);
					for(int i=head[a%mod];i!=-1;)//i不等于-1，是指可以在next数组里找到其他特征值相同的值 
					{
						if(hash[i]==a)
						{
							flag=1;
							break;
						}
						i=next[i];
					}	
					if(flag) 
					printf("YES\n");
					else
					printf("NO\n");
				}
			}
		}
	}
	return 0;
}