NYOJ 找球号(一)

 

描述

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

using namespace std;

vector<int> q;

//二分查找
void getAnswer(int aim,int m){

    int left = -1,right = m,mid;

    while(left + 1 < right) {
        mid = (left + right) >> 1;//向左移位1表示取中间值(二进制)
        //找到目标数字
        if(q[mid] == aim){
            printf("YES\n");
            break;
        }
        else if(q[mid] > aim)right = mid;//当去q[mid] > aim时 取[left,mid]
        else if(q[mid] < aim)left = mid;//当去q[mid] < aim时 取[mid,right]
    }
    //没有找到时
    if(left + 1 >= right)printf("NO\n");
}

int main(int argc, char const *argv[]) {
    //m-球数 n-查找数
    int m,n;

    scanf("%d%d",&m,&n);

    for(int i = 0;i < m;i++){
        int a;
        scanf("%d",&a);

        q.push_back(a);
    }

    sort(q.begin(),q.end());//排序

    for(int i = 0;i < n;i++){
        int b;
        scanf("%d",&b);
        
        getAnswer(b,m);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/WX_1218639030/article/details/84442125