NYOJ138:找球号(二)(Hash)

来源:互联网 发布:手机装潢设计软件 编辑:程序博客网 时间:2024/04/19 15:20

找球号(二)

时间限制: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".
样例输入
2ADD 5 34 343 54 6 2QUERY 4 34 54 33 66
样例输出
YESYESNONO
来源
[苗栋栋]原创
思路:用哈希表做,所谓哈希表,原理跟图论中的前向星结构差不多。

# include <iostream># include <cstdio># include <cstring># define maxn 1000000//开大一点,开小了就RE# define mod 111123using namespace std;int cnt, Hash[maxn+3], head[maxn+3], pre[maxn+3];void Add(int num){    int key = num % mod;    pre[cnt] = head[key];//记录上一个位置    head[key] = cnt;//更新指针    Hash[cnt++] = num;}bool query(int num){    int key = num % mod;    for(int i=head[key]; i>-1; i=pre[i])        if(Hash[i] == num)            return true;    return false;}int main(){    cnt = 0;    int n, m, num;    char s[10];    scanf("%d",&n);    memset(head, -1, sizeof(head));    while(n--)    {        scanf("%s%d",s,&m);        if(s[0] == 'A')            while(m--)            {                scanf("%d",&num);                Add(num);            }        else            while(m--)            {                scanf("%d",&num);                if(query(num))                    puts("YES");                else                    puts("NO");            }    }    return 0;}


0 0
原创粉丝点击