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

详解：
方法一：hash+邻接表
        http://www.cnblogs.com/yyroom/archive/2013/08/21/3271959.html
方法二：http://blog.youkuaiyun.com/qq523176585/article/details/17427469


hash+邻接表

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
 
#define N 1000001;
 
typedef struct node
{
    int data;
    node *next;
}node;
node *head[1000005];

void insert(int x)
{
    int ans = x % N;
    if(head[ans] == NULL)
    {
        head[ans] = (node *) malloc(sizeof(node));
        head[ans]->next = NULL;
        head[ans]->data = -1;
    }
 
    node *p;
    p = head[ans];
    int flag = 1;
    while(p->next != NULL)
    {
        if(p->next->data == x)
        {
            flag = 0;
            break;
        }
        p = p ->next;
    }
    node *q;
    if(flag)
    {
        q = (node *) malloc (sizeof(node));
        q -> data = x;
        q->next = NULL;
        p->next = q;
        p = q;
    }
}
 
int find(int x)
{
    int ans = x % N;
    node *p;
    if(head[ans] == NULL)
        return 0;
    else
    {
        p = head[ans]->next;
        while(p != NULL)
        {
            if(p->data == x)
                return 1;
            p = p->next;
        }
        return 0;
    }
}
int main()
{
    int x,t,T;
    scanf("%d",&T);
    char ch[12];
    while(T--)
    {
     scanf("%s",ch);
     if(ch[0] == 'A')
        {
            scanf("%d",&t);
            while(t--)
            {
                scanf("%d",&x);
                insert(x);
            }
        }
    else
    {
        scanf("%d",&t);
        while(t--)
        {
            scanf("%d",&x);
            if(find(x) == 1)
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    }
    return 0;
}

方法二：

 
#include <stdio.h>  
  
#define BITSPERWORD  32   /*一个整型数值的大小*/  
#define SHIFT        5    /* 2^5 = 32,即一个整型数值有32位，那么用一个元素就能保存32个状态位，0表示不存在，1表示存在*/  
#define MASK         0x1F /*低5位*/  
#define N            100000000 /*待处理的数字的个数*/  
  
int a[1+N/BITSPERWORD];/*节省大量的空间*/  
  
int main(void)  
{  
    int m,n,x;  
    char s[10];  
    scanf("%d%d",&m);  
    while (m--)  
    {  
        scanf("%s%d",s,&n);  
        if (s[0] == 'A')  
        {  
            while (n--)  
            {  
                scanf("%d",&x);  
                a[x>>SHIFT] |= (1<<(x & MASK));  
            }  
        }  
        else  
        {  
            while (n--)  
            {  
                scanf("%d",&x);  
                if (a[x>>SHIFT] & (1<<(x & MASK)))  
                {  
                    printf("YES\n");  
                }  
                else  
                {  
                    printf("NO\n");  
                }  
            }  
        }  
    }  
}