nyoj86 找球号(一）二分查找 set容器

找球号(一）

时间限制： 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<cstdio>
#include<algorithm>
int num[1000005];
using namespace std;

void fun(int k,int m)//在长度为m的数组num找到元素k. 
{
	int high=m-1,low=0,mid,flag=0;//flag做标记 
	
	while(low<=high)
	{
		mid=(low+high)/2;
		if(num[mid]<k)
		low=mid+1;
		else if(num[mid]>k)
		high=mid-1;
		else //num[mid]==k
		{
			flag=1;
			printf("YES\n");
			break;
		}
	}
	if(flag==0)	
	printf("NO\n"); 
}

int main()
{  
	int m,n,k;
	scanf("%d%d",&m,&n);
	for(int i=0;i<m;i++)
	scanf("%d",&num[i]);
	
	sort(num,num+m);//sort排序 
	while(n--)
	{
		scanf("%d",&k);
		fun(k,m);
	}
	return 0;
} 

第二种 set容器

#include<stdio.h>
#include<set>
using namespace std;
int main()
{
	int n,m,x,i;
	set<int>s;
	scanf("%d %d",&m,&n);
	for(i=0;i<m;i++)
	scanf("%d",&x),s.insert(x);
	for(i=0;i<n;i++)
	{
		scanf("%d",&x);
		if(s.find(x)!=s.end())//核心代码 
		printf("YES\n");
		else
		printf("NO\n");
	}
	return 0;
}