找球号-二分查找

1182 - 找球号(一）

时间限制：3秒 内存限制：128兆

题目描述在某一国度里流行着一种游戏。游戏规则为：在一堆球中，每个球上都有一个整数编号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<iostream>
#include<algorithm>
#define M 1000010
int a[M];
using namespace std;
int find(int b,int i,int j)
{
while(i<=j)
{
if(b>a[j]||b<a[i])
return 0;
int m=(i+j)/2;
if(b==a[m]) return 1;
else if(b>a[m]) i=m+1;
else j=m-1;
}
return 0;
}
int main()
{
int n,m,b;
cin>>n>>m;
for(int i=0;i<n;i++)
cin>>a[i];
sort(a,a+n);
for(int i=0;i<m;i++)
{
cin>>b;
if(find(b,0,n-1))
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
}