【二分查找】眼红的Medusa

题目

P1571 眼红的Medusa

---

题目描述

虽然 Miss Medusa 到了北京，领了科技创新奖，但是她还是觉得不满意。原因是：他发现很多人都和她一样获了科技创新奖，特别是其中的某些人，还获得了另一个奖项——特殊贡献奖。而越多的人获得了两个奖项，Miss Medusa就会越眼红。于是她决定统计有哪些人获得了两个奖项，来知道自己有多眼红。

输入格式

第一行两个整数 $n, m$，表示有 $n$ 个人获得科技创新奖，$m$ 个人获得特殊贡献奖。

第二行 $n$ 个正整数，表示获得科技创新奖的人的编号。

第三行 $m$ 个正整数，表示获得特殊贡献奖的人的编号。

输出格式

输出一行，为获得两个奖项的人的编号，按在科技创新奖获奖名单中的先后次序输出。

样例 #1

样例输入 #1

4 3
2 15 6 8
8 9 2

样例输出 #

2 8

提示

对于 $60\%$ 的数据，$0 \leq n, m \leq 1000$，获得奖项的人的编号 $\lt 2 \times 10^9$；

对于 $100\%$ 的数据，$0 \leq n, m \leq 10^5$，获得奖项的人的编号 $\lt 2 \times 10^9$。

输入数据保证第二行任意两个数不同，第三行任意两个数不同。

解题

方法一：哈希表

思路

维护哈希表统计同时获得两个奖的人，按照科技创新奖的输入顺序输出即可。

代码

#include <iostream>
#include <unordered_map>
#include <vector>

using namespace std;

int n, m;

int main() {
    scanf("%d%d", &n, &m);
    vector<int> a;
    unordered_map<int, int> cnts;
    while (n--) {
        int x;
        scanf("%d", &x);
        a.push_back(x);
        ++cnts[x];
    }
    while (m--) {
        int x;
        scanf("%d", &x);
        ++cnts[x];
    }
    for (auto& x : a) {
        if (cnts[x] == 2) printf("%d ", x);
    }

    return 0;
}

方法二：二分查找

思路

把得奖的人的编号分别输入到两个数组中，对获得特殊贡献奖的那个数组排序，然后遍历获得科技创新奖的所有编号，对于每一个获得科技创新奖的人在另一个数组中二分查找，如果找到就输出。

代码

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10;
int n, m;
int a[N], b[N];

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; ++i) scanf("%d", &a[i]);
    for (int i = 0; i < m; ++i) scanf("%d", &b[i]);
    sort(b, b + m);
    for (int i = 0; i < n; ++i) {
        int& x = a[i];
        int l = 0, r = m - 1;
        while (l < r) {
            int m = l + r >> 1;
            if (b[m] >= x) r = m;
            else l = m + 1;
        }
        if (b[l] == x) printf("%d ", x);
    }

    return 0;
}