NYOJ-138 找球号2 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

题目大意：

如题。

思路：

传送门    一个介绍hash的文章。

基本hash 的用法吧。

其中那个fib  是一个大素数，不一定是我写的这个数。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<time.h>
#include<algorithm>
using namespace std;

#define N 1000010
#define CLR(arr, what) memset(arr, what, sizeof(arr))

const int fib = 131313;
int Key[N], Head[N], Next[N];
int top;

void add(int n)
{
	int temp;
	temp = n % fib;
	Key[top] = n;
	Next[top] = Head[temp];
	Head[temp] = top;
	top++;
}

int main()
{
	int ncase;
	char str[8];
	int num, number;
	bool flag;
	CLR(Key, 0);
	CLR(Head, -1);
	top = 0;
	flag = false;
	scanf("%d", &ncase);
	while(ncase--)
	{
		scanf("%s", str);
		if(str[0] == 'A')
		{
			scanf("%d", &num);
			for(int i = 0; i < num; ++i)
			{
				scanf("%d", &number);
				add(number);
			}
		}
		else
		{
			scanf("%d", &num);
			for(int i = 0; i < num; ++i)
			{
				scanf("%d", &number);
				int temp = number % fib;
				for(int j = Head[temp]; j != -1; j = Next[j])
					if(Key[j] == number)
					{
						flag = true;
						break;
					}
					printf(flag == true ? "YES\n" : "NO\n");
					flag = false;
			}
		}
	}
	return 0;
}