计蒜客 找球号

在蒜国里流行着一种游戏。

游戏规则为：在一堆球中，每个球上都有一个整数编号 i(0≤i≤109)，编号可重复，现在说一个随机整数 k(0≤k≤109+100)，判断编号为 k 的球是否在这堆球中（存在为"YES"，否则为"NO"），先答出者为胜。现在有一个人想玩玩这个游戏，但他又很懒。他希望你能帮助他取得胜利。
输入格式

第一行有两个整数 m，n(0≤m≤106,0≤n≤106)；

m 表示这堆球里有 m 个球，n表示这个游戏进行 n 次。

接下来输入 m+n个整数，前 m 个分别表示这 m 个球的编号 i，后 n 个分别表示每次游戏中的随机整数 k。
输出格式

输出"YES"或"NO"。

样例输入

6 4
23 34 46 768 343 343
2 4 23 343

样例输出

NO
NO
YES
YES

思路：一开始使用map超时了，改用set即可，也可以存入数组排序后用二分查找

#include <cstdio>
//#include <map>
#include <set> 
using namespace std;
//map<int, int> mp;
set<int> st;
int main() {
	int m ,n;
//	cin>>m>>n;
	scanf("%d%d", &m, &n);
	for (int i = 0; i < m; i++ ) {
		int temp;
//		cin>>temp;
		scanf("%d", &temp);
//		mp[temp] = 1;
		st.insert(temp);
	}
	for (int i = 0; i < n; i++ ) {
		int t;
//		cin>>t;
		scanf("%d", &t);
		if (st.count(t)) {
//			cout<<"YES"<<endl;
			printf("YES\n");
		} else {
			printf("NO\n");
		}
	}
	return 0;
}