nyoj 找球号(二）

来源：互联网发布：知乎飞机杯编辑：程序博客网时间：2024/05/01 18:58

找球号(二）

在某一国度里流行着一种游戏。游戏规则为：现有一堆球中，每个球上都有一个整数编号i(0<=i<=100000000),编号可重复，还有一个空箱子，现在有两种动作：一种是"ADD"，表示向空箱子里放m(0<m<=100)个球，另一种是"QUERY”，表示说出M(0<M<=100)个随机整数ki(0<=ki<=100000100),分别判断编号为ki 的球是否在这个空箱子中（存在为"YES",否则为"NO"），先答出者为胜。现在有一个人想玩玩这个游戏，但他又很懒。他希望你能帮助他取得胜利。

Input

第一行有一个整数n(0<n<=10000);
随后有n行;
每行可能出现如下的任意一种形式：
第一种：
一个字符串"ADD"，接着是一个整数m，随后有m个i；
第二种：
一个字符串"QUERY”，接着是一个整数M，随后有M个ki；

Output

输出每次询问的结果"YES"或"NO".

Sample Input

2ADD 5 34 343 54 6 2QUERY 4 34 54 33 66

Sample Output

YESYESNONO

方法一（vector容器）：

#include<stdio.h>#include<string.h>#include<vector>#include<algorithm>using namespace std;vector<bool>a(100000001);int main(){    char s[10];    int t;    scanf("%d",&t);    getchar();    while(t--)    {        int n,i,m;        memset(s,'\0',sizeof(s));        scanf("%s",s);        scanf("%d",&n);        if(!strcmp(s,"ADD"))        {            while(n--)            {                scanf("%d",&m);                a[m]=true;            }        }        else if(!strcmp(s,"QUERY"))        {            while(n--)            {                scanf("%d",&m);                printf(a[m]?"YES\n":"NO\n");            }        }    }    return 0;}

ps：这种方法还是比较好用的。。

方法二（Hash表）：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<time.h>
#include<algorithm>
using namespace std;
#define N 1000010
#define CLR(arr, what) memset(arr, what, sizeof(arr))
const int fib = 111123;
int Key[N], Head[N], Next[N];
int top;
void add(int n)
{
int temp;
temp = n % fib;
Key[top] = n;
Next[top] = Head[temp];
Head[temp] = top;
top++;
}
int main()
{
int ncase;
char str[8];
int num, number;
bool flag;
CLR(Key, 0);
CLR(Head, -1);
top = 0;
flag = false;
scanf("%d", &ncase);
while(ncase--)
{
scanf("%s", str);
if(str[0] == 'A')
{
scanf("%d", &num);
for(int i = 0; i < num; ++i)
{
scanf("%d", &number);
add(number);
}
}
else
{
scanf("%d", &num);
for(int i = 0; i < num; ++i)
{
scanf("%d", &number);
int temp = number % fib;
for(int j = Head[temp]; j != -1; j = Next[j])
if(Key[j] == number)
{
flag = true;
break;
}
printf(flag == true ? "YES\n" : "NO\n");
flag = false;
}
}
}
return 0;
}

ps:这种方法对我等小白来说有点难，待我先提升代码能力后再水此法。。

方法三（位存储水过）：

#include<stdio.h>
unsigned int hash[3125005] = {0};
int main()
{
int i,a,t,n;
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);
hash[a / 32] |= 1 << (a % 32);
}
}
else if(str[0] == 'Q')
{
for(i = 0; i < n; ++i)
{
scanf("%d", &a);
if(hash[a / 32] & (1 << (a % 32)))
printf("YES\n");
else
printf("NO\n");
}
}
}
return 0;
}

ps:3125005怎么来的？因为最大值是10^7+10。用32来除法散列，（10^7+10）/32 ~~3125000.3125，所以取3125005。为什么是用32来散列呢？数据说了数值不会超过32位。

1 0