找球号(二）

找球号(二）

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

思路：

哈希表的应用， 用的是数组法，不知什么原因用链表法超时， 可能是数组访问速度较快吧。。。（就是数组法那三个数组挺绕的的）

/*
head[i]: i表示键值（num%MAX_NUM）， 存的是i在key中的位置
next[i]: i表示键值（num%MAX_NUM）在key中位置的下一个位置（类似链表）
key[i]: 存储原始数据
*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define MAX_NUM 100011

int key[1000001], next[1000001], head[1000001], top;

int inserthash(int num)       //将num插入哈希表
{
    int t = num % MAX_NUM;
    key[top] = num;
    next[top] = head[t];
    head[t] = top;
    top++;
    return 0;
}

int find(int num)            //查询是否已存在
{
    int t = num % MAX_NUM, i;
    for(i = head[t]; i != -1; i = next[i])
    {
        if(num == key[i])
        {
            return 1;
        }
    }
    return 0;
}

int main()
{
    char str[10];
    int n, m, i, temp;
    for(i = 0; i < 1000000; i++)   //初始化
    {
        head[i] = -1;
        key[i] = 0;
    }
    scanf("%d", &n);
    while(n--)
    {
        scanf("%s%d", str, &m);
        if(str[0] == 'A')
        {
            for(i = 0; i < m; i++)
            {
                scanf("%d", &temp);
                inserthash(temp);
            }
        }
        else
        {
            for(i = 0; i < m; i++)
            {
                scanf("%d", &temp);
                if(find(temp))
                {
                    printf("YES\n");
                }
                else
                {
                    printf("NO\n");
                }
            }
        }
    }
    return 0;
}