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

最新推荐文章于 2021-08-09 17:07:59 发布

原创最新推荐文章于 2021-08-09 17:07:59 发布 · 138 阅读

0 ·

CC 4.0 BY-SA版权

算法笔记专栏收录该内容

108 篇文章

订阅专栏

本文探讨了如何使用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;
}