NYOJ 138找球号（二）

/*找球号(二）

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



#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;


#define N 1000000
#define size 1000010

typedef struct node{
    int data;
    struct node *next;
}hash,*Hash;

hash head[size];

Hash search(hash head[size],int value)
{
    int k = value % N;
    Hash p = &head[k];
    while(p && p->data != value)
    {
        p = p->next;
    }
    return p;
}

void insert(hash head[size],int value)
{
    Hash s;
    int k = value % N;
    if(!search(head,value))
    {
        s = new node;
        s->data = value;
        s->next = NULL;
        s->next = head[k].next;
        head[k].next = s;
    }
}


int main()
{
    //freopen("in.txt","r",stdin);
    int T,n,i,value;
    char s[7];
    for(i=0; i<size; ++i)
    {
        head[i].next = NULL;
        head[i].data = -1;
    }
    scanf("%d",&T);
    while(T--)
    {
        cin>>s;
        scanf("%d",&n);
        for(i=0; i<n; ++i)
        {
            scanf("%d",&value);
            if(s[0] == 'A')
            {
                insert(head,value);    
            }
            else
            {
                if(search(head,value))
                    cout<<"YES\n";
                else
                    cout<<"NO\n";
            }
        }
    }
    return 0;
}

//用数组存储也可以，但是空间要求太大了，还可以用位存储来做，不知道怎么弄，也得学习一下……。