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

2024-02-09 15:48

文章标签 二分 hash 找球 nyoj138

本文主要是介绍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");elseprintf("NO\n");}}} return 0;
}