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<bits/stdc++.h>  
using namespace std;  
int main()  
{  
    vector<bool>pq(100000011);  
    int n,m,d;  
    char s[10];  
    scanf("%d",&n);  
    for(int i=0;i<n;i++)  
    {  
        scanf("%s%d",s,&m);  
        if(s[0]=='A')  
        {  
            for(int j=0;j<m;j++)  
            {  
                scanf("%d",&d);  
                pq[d]=true;  
            }  
        }  
        else  
        {  
            for(int j=0;j<m;j++)  
            {  
                scanf("%d",&d);  
                if(pq[d])  
                    printf("YES\n");  
                else  
                    printf("NO\n");  
            }  
        }  
    }  
    return 0;  
}          

//http://blog.csdn.net/kay_zhyu/article/details/8851850?locationNum=2&fps=1
//离散加哈希表,理解起来有难度 
#include<stdio.h>
#include<string.h>
#include<math.h>
const int N = 1000002;
#define MAX 200003;
int Head[N];
int Next[N];
int Hash[N];
void add(int &top, int num){ //int &top，必须将top引用，否则超时 
	int key = num%MAX;  
	//想向成链表，只是将链表转换成了数组来表示 
	Next[top] = Head[key];  //插入节点，那么这个节点指向的后面肯定为空 
	Head[key] = top;  //头结点连着这个插入的结点 
	Hash[top] = num;   //给这个结点一个值 
	++top;
}   //这一部分你也可以用作链表来存储 
bool search(int num){
	int key = num%MAX;
	for(int i = Head[key]; i > -1; i = Next[i]){  //依次寻找下一结点即可 
		if(Hash[i] == num){
			return true;
		}
	}
	return false;
}
int main(){
	int n, m, x;
	int top = 0;
	char str[10];
	scanf("%d", &n);
	//相当于链表中头结点为空 
	memset(Head, -1, sizeof(Head));
	while(n--){
		scanf("%s %d", str, &m);
		if(str[0] == 'A'){
			for(int i = 0; i < m; i++){
				scanf("%d", &x);
				add(top, x);
			}
		}
		else if(str[0] == 'Q'){
			for(int i = 0; i < m; i++){
				scanf("%d", &x);
				if(search(x))
					printf("YES\n");
				else
					printf("NO\n");
			}
		}
	}
	return 0;
}