NYOJ 138 找球号(二）

╰半橙微兮° 2022-09-19 08:19 167阅读 0赞

## 找球号(二） ##

时间限制： 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 <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");
    				}
    			}
    		}
    	}
    }