【王道机试】二分查找-查找-北邮

OJ：查找

描述
输入数组长度 n 输入数组 a[1…n] 输入查找个数m 输入查找数字b[1…m] 输出 YES or NO 查找有则YES 否则NO 。
输入描述：
输入有多组数据。 每组输入n，然后输入n个整数，再输入m，然后再输入m个整数（1<=m,n<=100）。
输出描述：
如果在n个数组中输出YES否则输出NO。
示例1
输入：

5
1 5 2 4 3
3
2 5 6

输出：

YES
YES
NO

本题采用二分查找的方式对目标元素进行查找，二分查找部分函数如下：

bool binary_search(int target,int length,int arr[]){
    int start=0,end=length-1;
    while(start<=end){
        int middle=(start+end)/2;
        if(arr[middle]<target){
            start=middle+1;
        }else if(arr[middle]>target){
            end=middle-1;
        }else{
            return true;
        }
    }
    return false;
}

要注意，二分查找仅适用于有序的数组，在使用二分查找前必须要对数组进行排序，对于上面实现的二分查找函数，需要数组为升序的有序数组。

总体代码如下：

#include<iostream>
#include<algorithm>
using namespace std;
bool binary_search(int target,int length,int arr[]){
    int start=0,end=length-1;
    while(start<=end){
        int middle=(start+end)/2;
        if(arr[middle]<target){
            start=middle+1;
        }else if(arr[middle]>target){
            end=middle-1;
        }else{
            return true;
        }
    }
    return false;
}
int main(){
    int n;
    while(cin>>n){
        int arr[n];
        for(int i=0;i<n;i++){
            cin>>arr[i];
        }
        sort(arr,arr+n);		//排序
        int m;
        cin>>m;
        while(m--){
            int target;
            cin>>target;
            if(binary_search(target,n,arr)){
                printf("YES\n");
            }else{
                printf("NO\n");
            }
        }
    }
    return 0;
}

另外，还可以使用vector数组中的 find 函数，具体可参考这篇文章：
vector<>-【王道机试】-查找-查找-北邮