NYOJ 138 找球号(二）

描述

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

分析：简单哈希。

Code：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <queue>
#include <cmath>
#include <map>
#include <set>
#define eps 1e-7
#define LL long long
#define pb push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;

const int MOD=1000007;
int Hash[MOD],head[MOD],next[MOD];
char s[10];

int main()
{
    int T,n,x,top=0;
    scanf("%d",&T);
    memset(Hash,-1,sizeof(Hash));
    memset(head,-1,sizeof(head));
    while(T--){
        scanf("%s %d",s,&n);
        if(s[0]=='A'){
            while(n--){
                scanf("%d",&x);
                int tmp=x%MOD;
                Hash[top]=x;
                next[top]=head[tmp];//拉链法处理冲突
                head[tmp]=top++;
            }
        }
        else {
            while(n--){
                scanf("%d",&x);
                int tmp=x%MOD;
                bool flag=false;
                for(int i=head[tmp];i!=-1;i=next[i]){
                    if(Hash[i]==x){
                        flag=true;
                        break;
                    }
                }
                if(flag) printf("YES\n");
                else printf("NO\n");
            }
        }
    }
    return 0;
}