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

找球号(二）

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

样例输入

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

样例输出

YESYESNONO

#include<cstdio> #include<string> #include<iostream>#include<cstring>const int maxn = 1000005,fib = 500000;using namespace std;int Hush[maxn],next[maxn],head[maxn];int top;int query (int key){int p=key%fib;for(int i=head[p];i>-1;i=next[i])if(Hush[i]==key)return 1;return 0;}void add(int num){int key=num%fib;next[top]=head[key];head[key]=top;Hush[top]=num;top++;}int main(){int t,n,i,num;    string s;    scanf ( "%d", &t );    top = 0;    memset ( head, -1, sizeof ( head ) );    while ( t -- )    {        cin >> s >> n;        switch ( s[0] )        {            case 'A' :                for ( i = 0; i < n; i ++ )                {                    scanf ( "%d", &num );                    add ( num );                }                break ;            case 'Q' :                for ( i = 0; i < n; i ++ )                {                    scanf ( "%d", &num );                    printf ( query ( num ) ? "YES\n" : "NO\n" );                }        }    }    return 0;}