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

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<>-【王道机试】-查找-查找-北邮

上一篇:数组的查找(搜索):线性查找和二分法查找


下一篇:FSF 称 DRM 被用于锁定、控制和监视用户