《算法竞赛进阶指南》 电影

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

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

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

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

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

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

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

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

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

输入格式

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

第二行输入 nn 个整数 a1,a2…ana1,a2…an，其中 aiai 表示第 ii 个科学家懂得的语言的编号。

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

第四行输入 mm 个整数 b1,b2…bmb1,b2…bm，其中 bibi 表示第 ii 部电影的语音采用的语言的编号。

第五行输入 mm 个整数 c1,c2…cmc1,c2…cm，其中 cici 表示第 ii 部电影的字幕采用的语言的编号。

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

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

输出格式

输出一个整数，代表最终选择的电影的编号。电影编号 1∼m1∼m。

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

数据范围

1≤n,m≤2000001≤n,m≤200000,
1≤ai,bi,ci≤1091≤ai,bi,ci≤109

输入样例：

3
2 3 2
2
3 2
2 3

输出样例：

2

解题思路以及代码：

//先补充3个函数，均在algorithm库中
//lower_bound(a.begin(), a.end(), x)返回a中大于等于x的第一个位置
//upper_bound(a.begin(), a.end(), x)返回a中大于x的第一个位置
//binary_search(a.begin(), a.end(), x)寻找a中的x若找到返回true否则返回false
/*
解题思路：虽然1e9中语言，但是n, m为1e5，n个人m场电影最多只有2 * m + n种语言
为1e5级别，所以我们需要用离散化将1e9映射到1e5
*/

时间复杂度：O(（n + m）log(n + m))

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 200010;

int n, m;
int a[N], b[N], c[N];
int lang[N * 3], uni[N * 3], cntl, cntu, ans[N * 3];//lang为稀疏数组，离散化后的uni为稠密数组。

int find(int x)//find作用是把稀疏图中的语言转化为映射后稠密图所在的坐标
{
    return lower_bound(uni + 1, uni + cntu + 1, x) - uni;
}

int main()
{
    cin >> n;//先将出现的所有语言的种类2 * n + m读入
    for (int i = 1; i <= n; i ++ )//将科学家所会的语言读入
    {
        scanf("%d", &a[i]);
        lang[ ++ cntl] = a[i];
    }

    cin >> m;
    for (int i = 1; i <= m; i ++ )//将科学界能听懂的语音读入
    {
        scanf("%d", &b[i]);
        lang[ ++ cntl] = b[i];
    }

    for (int i = 1; i <= m; i ++ )//将科学家能看懂的字幕读入
    {
        scanf("%d", &c[i]);
        lang[ ++ cntl] = c[i];
    }

    sort(lang + 1, lang + cntl + 1);//排序是为了去掉lang中的重复元素，进而转化为稠密图
    //sort为时间复杂度的瓶颈
    for (int i = 1; i <= cntl; i ++ )//将lang去重转化为稠密组数uni
    {
        if (i == 1 || lang[i] != lang[i - 1])
            uni[++ cntu] = lang[i];
    }

    for (int i = 1; i <= n; i ++ )//记录各个科学家掌握的语言在稠密图中的位置，
        ans[find(a[i])] ++ ;      //可以得出哪种语言科学家掌握的最多

    int ans0, ans1, ans2;//ans0为最终的答案，ans1为科学家能听懂语音的人数
    ans0 = ans1 = ans2 = 0;//ans2为科学家能看懂字幕的人数

    for (int i = 1; i <= m; i ++ )
    {
        int ansx = ans[find(b[i])], ansy = ans[find(c[i])];
        //若次循环听得懂语言的科学家大于之前循环过的听得懂语言的科学家的人数
        //或者听得懂语音的科学家等于之前循环过的听得懂语音的科学家的人数
        //但是这次循环看的懂字母的科学家大于之前循环过的看得懂字母的科学家的人数
        if (ansx > ans1 || (ansx == ans1 && ansy > ans2))
        {
            ans0 = i, ans1 = ansx, ans2 = ansy;
        }
    }

    if (ans0 == 0)//说明没用科学家看得懂的语音和听的懂得字幕随便输出一个
        cout << 1 << endl;
    else cout << ans0 << endl;

    return 0;
}