nyoj-138-找球号（二）----hash算法之除留余数法+vector

找球号(二）

时间限制：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>
#include<string.h>
#include<vector>
using namespace std;
#define M 12345
vector<int> hash[M];
void ADD(int x)
{
int h=x%M;
hash[h].push_back(x);//把每个取余之后的值压入vector;
}
void Query(int x)
{
int h=x%M;
bool flag=false;
for(int i=0;i<hash[h].size();i++)//每次取余之后查找此余数所存的值
{
if(x==hash[h][i])
{
flag=true;printf("YES\n");break;
}
}
if(!flag) printf("NO\n");
}
int main()
{
int N,n,m,i,a,b;
char str[10];
scanf("%d",&N);
while(N--)
{
scanf("%s",str);
if(strcmp(str,"ADD")==0)
{
getchar();
scanf("%d",&n);
for(i=0;i<n;i++)
{
scanf("%d",&a);
ADD(a);
}
}
else
{
scanf("%d",&m);
for(i=0;i<m;i++)
{
scanf("%d",&b);
Query(b);
}
}
}
}

在网上看到一个比较优秀的代码，但有点看不懂，请大神指教，如下：

#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;
}