描述
在某一国度里流行着一种游戏。游戏规则为:在一堆球中,每个球上都有一个整数编号i(0<=i<=100000000),编号可重复,现在说一个随机整数k(0<=k<=100000100),判断编号为k的球是否在这堆球中(存在为"YES",否则为"NO"),先答出者为胜。现在有一个人想玩玩这个游戏,但他又很懒。他希望你能帮助他取得胜利。
输入
第一行有两个整数m,n(0<=n<=100000,0<=m<=1000000);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
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> q;
//二分查找
void getAnswer(int aim,int m){
int left = -1,right = m,mid;
while(left + 1 < right) {
mid = (left + right) >> 1;//向左移位1表示取中间值(二进制)
//找到目标数字
if(q[mid] == aim){
printf("YES\n");
break;
}
else if(q[mid] > aim)right = mid;//当去q[mid] > aim时 取[left,mid]
else if(q[mid] < aim)left = mid;//当去q[mid] < aim时 取[mid,right]
}
//没有找到时
if(left + 1 >= right)printf("NO\n");
}
int main(int argc, char const *argv[]) {
//m-球数 n-查找数
int m,n;
scanf("%d%d",&m,&n);
for(int i = 0;i < m;i++){
int a;
scanf("%d",&a);
q.push_back(a);
}
sort(q.begin(),q.end());//排序
for(int i = 0;i < n;i++){
int b;
scanf("%d",&b);
getAnswer(b,m);
}
return 0;
}