NYOJ - 找球号(一）(二分)

找球号(一）

时间限制： 3000 ms  |           内存限制： 65535 KB

难度： 3

描述

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

using namespace std;

int a[1000001];

bool search(int index,int m)
{
	int low = 0;
	int high = m -1;
	int mid;
	while(low <= high)
	{
		mid = (low + high)/2;
		if(index < a[mid])
		high = mid -1;
		else if(index > a[mid])
		low = mid + 1;
		else
		return 1;
	}
	return 0;
}
int main()
{
	int m,n,i,k;
	scanf("%d%d",&m,&n);
	for(i = 0; i < m; i++)
	{
	   scanf("%d",&a[i]);
	}
	sort(a,a+m);
	for(i = 0; i < n; i++)
	{
	   scanf("%d",&k);
	   printf("%s\n",search(k,m)?"YES":"NO");
	}
	return 0;
}        

这题主要用二分查找来解决，注意，只要输入一组需要处理的数据，还有输入输出的优化，不然会超时的。这题已经AC，这里提供参考。