NYOJ 找球號(一)
阿新 • • 發佈:2018-11-29
描述
在某一國度裡流行著一種遊戲。遊戲規則為:在一堆球中,每個球上都有一個整數編號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; }