找球号(二）

找球号(二）

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

分析：

题目一读完就觉得很简单，直接有set判断是否存在嘛，但是超时了。对于STL的使用，以前也只是随便看了看，看来接下来要好好的研究一下了。

看了看别人的博客，知道了用哈希表。（也不是很熟，不过已经找到比较好的书了，相信下次会好些的）。在hash[]中存的是各个输入的数，Head[]存每个key值得头“指针“，next[]指向下一个具有相同key值的数。

哈希参考资料：http://blog.csdn.net/v_JULY_v/article/details/6256463（向大神学习）

#include<cstdio>
#include<cstring>
const int N = 1e6 + 10;
#define mod 200003
int Head
, Next
, Hash
;
int top;

bool find(int a)
{
bool flag = false;
for(int i = Head[a%mod]; i != -1; i = Next[i])
{
if(Hash[i] == a)
{
flag = true;
break;
}
}
return flag;
}

void add(int num)
{
if(!find(num))
{
int key = num % mod;
Hash[top] = num;
Next[top] = Head[key];
Head[key] = top;
top++;
}
}

int main()
{
int n, m, a;
char op[10];
while(~scanf("%d",&n)) {
memset(Head, -1, sizeof(Head));
top = 0;
while(n--) {
scanf("%s%d", op, &m);
if(op[0] == 'A') {
for(int i = 0; i < m; i++) {
scanf("%d",&a);
add(a);
}
}
else {
while(m--) {
int flag = 0;
scanf("%d",&a);
if(find(a)) printf("YES\n");
else printf("NO\n");
}
}
}
}
return 0;
}