NYOJ138 找球号（二）

 

找球号(二）

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

//代码一：用邻接表存储--hash
#include<stdio.h>
#include<string.h>
#include<malloc.h>
#define Mod 10000
struct Node
{
    int num;
    struct Node *next;
}node[10001];

int main()
{     
    int n,t,temp,i;
    char s[10];
    struct Node *t1;
    scanf("%d",&n);
    for(i=0;i<10001;++i)
    {
        node[i].num=i;
        node[i].next=NULL;
    }
    while(n--)
    {    
        scanf("%s",s);
        if(s[0]=='A')
        {
            scanf("%d",&t);
            for(i=0;i<t;++i)
            {
                scanf("%d",&temp);
                t1=(struct Node *)malloc(sizeof(struct Node));
                t1->num=temp;
                t1->next=node[temp%Mod].next;
                node[temp%Mod].next=t1;
            }
        }
        else
        {
            scanf("%d",&t);
            for(i=0;i<t;++i)
            {
                scanf("%d",&temp);
                t1=node[temp%Mod].next;
                if(t1==NULL)
                {
                    printf("NO\n");
                    continue;
                }
                while(t1)
                {
                    if(t1->num==temp)
                    {
                        printf("YES\n");
                        break;
                    }
                    t1=t1->next;    
                }
                if(t1==NULL)
                    printf("NO\n");
            }
        }    
    }
    return 0;
}

//代码二：利用二进制位 hash
#include <iostream>
#include <cstdio>

using namespace std;

int flag[3125001];

int main()
{
    int n, t;
    scanf("%d", &n);
    char cmd[6];
    while(n--)
    {
        scanf("%s", cmd);
        scanf("%d", &t);
        int temp;
        if(cmd[0] == 'A')
        {
            while(t--)
            {
                scanf("%d", &temp);
                flag[temp / 32] |= 1 << (temp % 32);
            }
        }
        else
        {
            while(t--)
            {
                scanf("%d", &temp);
                if(flag[temp / 32] & 1 << (temp % 32))
                    printf("YES\n");
                else
                    printf("NO\n");
            }
        }
    }
    return 0;
}