1299: 【二分查找】找球号

题目描述

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

一道二分查找题

代码：

#include <bits/stdc++.h>
using namespace std;
int a[1000005];
int main()
{
    int n,m,i,mid,l,r,num;
    cin>>n>>m;
    for(i=0;i<n;i++)
        scanf("%d",&a[i]);
        sort(a,a+n);//升序数列
    while(m--)
    {
        scanf("%d",&num);
        l=-1;r=n;
        while(l+1!=r)
        {
            mid=(l+r)/2;
            if(a[mid]<num) l=mid;
            if(a[mid]>=num) r=mid;
        }
        if(a[r]==num) printf("YES\n");//a[r]为第一个>=num的数；
        else printf("NO\n");
    }//二分查找
    return 0;
}

 还可以用STL的lower_bound()函数写，查找第一个大于等于key的值，其返回值是迭代器；upper_bound()返回第一个大于key的值。如果数组中不存在大于key的值，则返回最后一个元素的下一个，即n;

lower_bound(arr[],arr[]+n,key)-arr[]为下标；

代码：

#include <bits/stdc++.h>
using namespace std;
int a[1000005];
int main()
{
    int n,m,ans;
    cin>>n>>m;
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
        sort(a,a+n);
    while(m--)
    {
        int k;
        scanf("%d",&k);
        ans=lower_bound(a,a+n,k)-a;
        if(a[ans]==k) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}