【牛客】查找（C++）

题目描述

输入数组长度 n 输入数组      a[1...n] 输入查找个数m 输入查找数字b[1...m]   输出 YES or NO  查找有则YES 否则NO 。

输入描述:

输入有多组数据。
每组输入n，然后输入n个整数，再输入m，然后再输入m个整数（1<=m,n<=100）。

输出描述:

如果在n个数组中输出YES否则输出NO。

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

int n;
int num[100];

bool find(int x)  //二分查找
{
    int left=0,right=n-1;
    while(left<=right)
    {
        int mid=left+(right-left)/2;  //防止溢出
        if(num[mid]>x)
            left=mid+1;
        else if(num[mid]>x)
            right=mid-1;
        else if(num[mid]==x)
            return 1;
    }
    return 0;
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
        {
            scanf("%d",&num[i]);
        }
        sort(num,num+n);  //排序
        int m;
        scanf("%d",&m);
        for(int j=0;j<m;j++)
        {
            int x;
            scanf("%d",&x);
            if(find(x))
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    return 0;
}