NYOJ 138 找球号



题目链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=138

题目描述：

描述

在某一国度里流行着一种游戏。游戏规则为：现有一堆球中，每个球上都有一个整数编号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

题目分析：这里只需要在最短的时间内完成查找的任务，那么无疑，我们设计的算法的时间复杂度最好为O(1)。这里很容易可以确定用HashTable来存储球的编号。对于hash表的详细介绍可在blog：http://blog.youkuaiyun.com/cyongxue/article/details/19544107。那么，假设每个hash表至多存放50个球，而球的总数为10^8个，那么可以创建10^8/50= 2*10^6个hash表，故mod2*10^6即可将球映射到在应的hash表中。

代码如下：

// 找球号.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"

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

typedef struct HashNode
{
	int nValue;
	struct HashNode *next;
}HashNode;

HashNode *hashTable[N];
const int N = 1000002;
const unsigned int MAX = 200003;

int Hash[N];
int Head[N];//相当于每一个散列表的头节点
int Next[N];//当前节点的下一个节点

void Hash_Init()
{
	memset(hashTable,NULL,N);
}
void Add(int num)
{
	int key = num % MAX;	
	HashNode *pHead = hashTable[key];
	HashNode *tmp = pHead->next;
	while(tmp)
	{
		if(tmp->nValue == num)
			return;
		else
			tmp = tmp->next;
	}
	HashNode *newNode = new HashNode;
	newNode->nValue = num;
	newNode->next = pHead->next;
	pHead->next = newNode;
}

bool Query(int num)
{
	int key = num % MAX;
	bool isExisted = false;
	HashNode *pHead = hashTable[key];
	HashNode *tmp = pHead->next;
	while(tmp)
	{
		if(tmp->nValue == num)
			isExisted = true;
		else
			tmp = tmp->next;
	}
	return isExisted;
}

int main()
{
	int t,i,n,a;
	Hash_Init();
	char str[10];
	scanf("%d",&t);
	while(t--)
	{
		scanf("%s %d", &str, &n);
		if(str[0] == 'A')
		{
			for(i = 0; i < n; ++i)
			{
				scanf("%d", &a);
				Add(a);
			}
		}
		else if(str[0] == 'Q')
		{
			for(i = 0; i < n; ++i)
			{
				scanf("%d", &a);
				if(Query(a))
					printf("YES\n");
				else
					printf("NO\n");
			}
		}//end if
	}//end while
	return 0;
}