NYOJ138：找球号(二）（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

来源

[苗栋栋]原创

思路：用哈希表做，所谓哈希表，原理跟图论中的前向星结构差不多。

# include <iostream>
# include <cstdio>
# include <cstring>
# define maxn 1000000//开大一点，开小了就RE
# define mod 111123
using namespace std;
int cnt, Hash[maxn+3], head[maxn+3], pre[maxn+3];

void Add(int num)
{
    int key = num % mod;
    pre[cnt] = head[key];//记录上一个位置
    head[key] = cnt;//更新指针
    Hash[cnt++] = num;
}

bool query(int num)
{
    int key = num % mod;
    for(int i=head[key]; i>-1; i=pre[i])
        if(Hash[i] == num)
            return true;
    return false;
}

int main()
{
    cnt = 0;
    int n, m, num;
    char s[10];
    scanf("%d",&n);
    memset(head, -1, sizeof(head));
    while(n--)
    {
        scanf("%s%d",s,&m);
        if(s[0] == 'A')
            while(m--)
            {
                scanf("%d",&num);
                Add(num);
            }
        else
            while(m--)
            {
                scanf("%d",&num);
                if(query(num))
                    puts("YES");
                else
                    puts("NO");
            }
    }
    return 0;
}