idea

题目描述

莫斯科正在举办一个大型国际会议，有n个来自不同国家的科学家参会。

每个科学家都只懂得一种语言。

为了方便起见，我们把世界上的所有语言用1到109之间的整数编号。

在会议结束后，所有的科学家决定一起去看场电影放松一下。

他们去的电影院里一共有m部电影正在上映，每部电影的语音和字幕都采用不同的语言。

对于观影的科学家来说，如果能听懂电影的语音，他就会很开心；如果能看懂字幕，他就会比较开心；如果全都不懂，他就会不开心。

现在科学家们决定大家看同一场电影。

请你帮忙选择一部电影，可以让观影很开心的人最多。

如果有多部电影满足条件，则在这些电影中挑选观影比较开心的人最多的那一部。

输入格式
第一行输入一个整数n，代表科学家的数量。

第二行输入n个整数a1,a2…an,其中ai表示第i个科学家懂得的语言的编号。

第三行输入一个整数m，代表电影的数量。

第四行输入m个整数b1,b2…bm,其中bi表示第i部电影的语音采用的语言的编号。

第五行输入m个整数c1,c2…cm,其中ci表示第i部电影的字幕采用的语言的编号。

请注意对于同一部电影来说，bi≠ci。

同一行内数字用空格隔开。

输出格式
输出一个整数，代表最终选择的电影的编号。

如果答案不唯一，输出任意一个均可。

数据范围
1≤n,m≤200000,
1≤ai,bi,ci≤109

输入样例：

3
2 3 2
2
3 2
2 3

输出样例：

2

算法1

时间复杂度 $O((n + m)log(n + m))$

参考文献 照抄的代码 QAQ

C++ 代码

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 2e5 + 10;
int n, m;
// a[]保存科学家懂得的语言的编号
// x[]保存电影的语音采用的语言的编号
// y[]电影的字幕采用的语言的编号
int a[N], x[N], y[N]; 
// idx数组c[]的索引
// cnt数组c去重后不重复元素的个数
// c[]保存a[]，x[]以及y[]中元素的数组
// ans[]统计开心度的数组 
int idx, cnt, c[N * 3], ans[N * 3];

int lBound(int x) {
    int l = 0, r = cnt - 1;
    while(l < r) {
        int m = l + r >> 1;
        if(x <= c[m]) r = m;
        else l = m + 1;
    }
    return l;
}

void discrete() {
    sort(c, c + idx);
    cnt = unique(c, c + idx) - c;
}

void input(int* v, int n) {
    for(int i = 0; i < n; i++) {
        cin >> v[i];
        c[idx++] = v[i];
    }
}

int main() {
    cin >> n;
    input(a, n);
    cin >> m;
    input(x, m); 
    input(y, m);

    discrete();

    // 统计对每部电影开心度的人数
    for(int i = 0; i < n; i++) ans[lBound(a[i])]++;

    int res = 0;
    int resx = 0, resy = 0; // 最开心的人数，比较开心的人数
    // 枚举m部电影的编号，获取每部电影让科学家开心的人数以及比较开心的人数
    for(int i = 0; i < m; i++) {   
        int h1 = ans[lBound(x[i])]; // 轮询最开心的人数 
        int h2 = ans[lBound(y[i])]; // 轮询比较开心的人数
        // 最开心的人数最多(不存在多个最多开心人数相等的情况)
        // 或者存在多个最多开心人数相等的情况就取比较开心的人数最多
        if(h1 > resx || (h1 == resx && h2 > resy)) {
            res = i + 1;
            resx = h1;
            resy = h2;
        }
    }
    cout << res << endl;
    return 0;
}