《算法笔记》4.5小节——算法初步-＞二分 问题 C: 查找

问题 C: 查找

时间限制 : 1.000 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

调用C++的二分查找函数

#include<iostream>
#include <algorithm>

using namespace std;


int main() {
    int n;
    int a[105];
    while (cin >> n) {
        int m;
        int b;
        for (int i = 0; i < n; ++i) {
            cin >> a[i];
        }
        sort(a, a + n);
        cin >> m;
        for (int i = 0; i < m; ++i) {
            cin >> b;
            int ans = lower_bound(a, a + n, b) - a;
            if (a[ans] != b)
                puts("NO");
            else
                puts("YES");
        }
    }
    return 0;
}

自己实现二分

#include<iostream>
#include <algorithm>

using namespace std;
int a[105];
int n;

int find(int l, int r, int x) {
    while (l < r) {
        int mid = l + r >> 1;
        if (a[mid] < x) l = mid + 1;
        else
            r = mid;
    }
    return l;
}

int main() {


    while (cin >> n) {
        int m;
        int b;
        for (int i = 0; i < n; ++i) {
            cin >> a[i];
        }
        sort(a, a + n);
        cin >> m;
        for (int i = 0; i < m; ++i) {
            cin >> b;
            int ans = find(0, n - 1, b);
            if (a[ans] != b)
                puts("NO");
            else
                puts("YES");
        }
    }
    return 0;
}

哈希实现效率显然更高

#include<iostream>
#include <unordered_set>

using namespace std;
int a[105];
int n;


int main() {


    while (cin >> n) {
        int m;
        int b;
        unordered_set<int> set;
        for (int i = 0; i < n; ++i) {
            cin >> a[i];
            set.insert(a[i]);
        }
        cin >> m;
        for (int i = 0; i < m; ++i) {
            cin >> b;
            if (set.count(b) == 0)
                puts("NO");
            else
                puts("YES");
        }
    }
    return 0;
}