Coder Pat之路 问题 C: 查找

问题 C: 查找

时间限制: 1 Sec   内存限制: 32 MB

题目描述

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

输入

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

输出

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

样例输入

6
3 2 5 4 7 8
2
3 6

样例输出

YES
NO

 
 
User: 吴锦诚
Date: 2018/6/10
#include <iostream>
#include <algorithm>
using namespace std;
bool cmp( int a, int b) {
     return a < b;
}
int main()
{
     //输入n
     int  n, an[101], m, bm[101], i, left, right, middle;
     while (cin>>n) {
         //输入数组an
         for (i = 1; i <= n; i++) {
             cin >> an[i];
         }
         //输入查找个数
         cin >> m;
         //输入查找数组bn
         for ( i = 1; i <=m; i++)
         {
             cin >> bm[i];
         }
         //排序
         sort(an+1, an + n+1, cmp);
         //输出YES or NO
         //遍历bm
         for (i = 1; i <= m; i++) {
             //二分遍历an
             left = 1;
             right = n;
             while (right >= left) {
                 middle = left + (right - left) / 2;
                 //二分分左边
                 if (an[middle] > bm[i]) {
                     right = middle - 1;
                 }
                 else if (an[middle]<bm[i]){
                     left = middle + 1;
                 }
                 else
                 {
                     cout << "YES" << endl;
                     break ;
                 }
             }
             if (right < left) {
                 cout << "NO" << endl;
             }
         }
     }
     return 0;
}
/**************************************************************
     Problem: 1937
     User: morizunzhu
     Language: C++
     Result: 正确
     Time:0 ms
     Memory:2024 kb
****************************************************************/

猜你喜欢

转载自blog.csdn.net/morizunzhu/article/details/80637874