nyoj138_找球号（二）（哈希表）

找球号(二）

时间限制： 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

哈希表中的链地址法。

#include<stdio.h>
#include<string.h>
#include<vector>
#define MOD 1000000
using namespace std;
struct Node
{
	int x;
	Node *next;
}*h[1000010];
void init()
{
	for(int i=0;i<=1000000;i++)
	{
		h[i]=new Node();
		h[i]->next=NULL;
	}
}
void add(int x)
{
	int key=x%MOD;
	Node *p=h[key];
	Node *q=new Node();
	q->next=NULL;
	q->x=x;
	while(p->next!=NULL)
		p=p->next;
	p->next=q;	
}
bool find(int x)
{
	int key=x%MOD;
	Node *p=h[key];
	while(p!=NULL)
	{
		if(p->x==x)
			return true;
		p=p->next;
	}
	return false;
}
int main()
{
	int i,j,t,n,x;
	char str[10];
	init();
	scanf("%d",&t);
	while(t--)
	{
		
		scanf("%s",str);
		if(str[0]=='A')
		{
			scanf("%d",&n);
			for(i=0;i<n;i++)
			{
				scanf("%d",&x);
				add(x);
			}
		}
		else
		{
			scanf("%d",&n);
			for(i=0;i<n;i++)
			{
				scanf("%d",&x);
				if(find(x))
					printf("YES\n");
				else
					printf("NO\n");
			}
			
		}
	}
	return 0;
}