1. 程式人生 > >NYOJ 找球號(一)

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;
}