NYOJ 138 找球号(二) bitset 二进制的妙用

找球号(二)

时间限制: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
来源

思路:直接开数组放不下,MLE,用bitset开可以放的下

代码:

#include<bits/stdc++.h>
using namespace std;
bitset<100000001> bs;
int main() {
int T;
scanf("%d",&T);
while(T--) {
char s[8];
int n,num;
scanf("%s",&s);
if(s[0]=='A') {
scanf("%d",&n);
for(int i=0;i<n;++i) {
scanf("%d",&num);
bs[num]=1;
}
} else {
scanf("%d",&n);
for(int i=0;i<n;++i) {
scanf("%d",&num);
if(bs[num]==1) {
printf("YES\n");
} else {
printf("NO\n");
}
}
}
}
return 0;
}

上一篇:C语言程序设计50例(三)(经典收藏)


下一篇:cs231n spring 2017 lecture4 Introduction to Neural Networks 听课笔记