0
点赞
收藏
分享

微信扫一扫

nyoj138 找球号(二) (hash表+二分)


题目链接:​​http://acm.nyist.net/JudgeOnline/problem.php?pid=138​​​
描述
在某一国度里流行着一种游戏。游戏规则为:现有一堆球中,每个球上都有一个整数编号i(0<=i<=100000000),编号可重复,还有一个空箱子,现在有两种动作:一种是”ADD”,表示向空箱子里放m(1<=m<=100)个球,另一种是”QUERY”,表示说出M(1<=M<=100)个随机整数ki(0<=ki<=100000100),分别判断编号为ki 的球是否在这个空箱子中(存在为”YES”,否则为”NO”),先答出者为胜。现在有一个人想玩玩这个游戏,但他又很懒。他希望你能帮助他取得胜利。
输入
第一行有一个整数n(1<=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

一开始用STL里面的set,超时了。后来才听说要用哈希表
使用set容器果然超时了
然后使用trie树 果然又超内存了
最后使用了hash表+二分 总算在内存不太大 时间不太长的情况下a了

#include <stdio.h>
#include <vector>
#include <algorithm>
#include <string.h>
using namespace std;
#define MOD 10000
vector<int>hash[MOD+10];
void insertHash(int x)
{
int pos=x%MOD;
hash[pos].insert(upper_bound(hash[pos].begin(),hash[pos].end(),x),x);
}
bool findHash(int x)
{
int pos=x%MOD;
int i=upper_bound(hash[pos].begin(),hash[pos].end(),x)-hash[pos].begin();
if(i==0)
return false;
return hash[pos][i-1]==x;
}
int main()
{
int n;
memset(hash,0,sizeof(hash));
scanf("%d",&n);
while(n--)
{
char type[15];
int x,m;
scanf("%s",type);
if(strcmp(type,"ADD")==0)
{
scanf("%d",&m);
while(m--)
{
scanf("%d",&x);
insertHash(x);
}
}
else
{
scanf("%d",&m);
while(m--)
{
scanf("%d",&x);
if(findHash(x))
printf("YES\n");
else
printf("NO\n");
}
}
}
return 0;
}


举报

相关推荐

0 条评论