1299: 【二分查找】找球号

1299: 【二分查找】找球号

时间限制: 1 Sec   内存限制: 128 MB

题目描述

在某一国度里流行着一种游戏。游戏规则为：在一堆球中，每个球上都有一个整数编号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"

样例输入 Copy

6 4

23 34 46 768 343 343

2 4 23 343

样例输出 Copy

NO

NO

YES

YES

解析：标题都说用二分查找就用二分查找做即可。这里给出非STL的解法。

#include<bits/stdc++.h>

using namespace std;

int read(){

int x=0,f=0;char c=getchar();

while(!isdigit(c)){if(c=='-')f=1;c=getchar();}

while(isdigit(c)){x=x*10+c-'0';c=getchar();}

return f?-x:x;

}

int a[1000005];

int n,m;

void init(){

n=read();m=read();

for(int i=1;i<=n;i++) a[i]=read();

sort(a+1,a+1+n);

}

int main(){

init();

for(int i=1;i<=m;i++){

int q=read();

if(q<a[1]||q>a[n]) printf("NO\n");

else{

int left=1,right=n;

int mid=(left+right)/2;

bool f=0;

while(left<=right){

// cout<<q<<":"<<left<<" "<<right<<" "<<a[mid]<<endl;

if(a[mid]>q) right=mid-1;

else if(a[mid]<q) left=mid+1;

else{f=1;break;}

mid=(left+right)/2;

}

if(a[left]==q||a[right]==q) f=1;

if(f) printf("YES\n");

else printf("NO\n");

}

}

return 0;

}