AcWing_105电影 C++题解

题目地址

题目描述

莫斯科正在举办一个大型国际会议，有n个来自不同国家的科学家参会。
每个科学家都只懂得一种语言。
为了方便起见，我们把世界上的所有语言用$1$到$10^9$之间的整数编号。
在会议结束后，所有的科学家决定一起去看场电影放松一下。
他们去的电影院里一共有m部电影正在上映，每部电影的语音和字幕都采用不同的语言。
对于观影的科学家来说，如果能听懂电影的语音，他就会很开心；如果能看懂字幕，他就会比较开心；如果全都不懂，他就会不开心。
现在科学家们决定大家看同一场电影。
请你帮忙选择一部电影，可以让观影很开心的人最多。
如果有多部电影满足条件，则在这些电影中挑选观影比较开心的人最多的那一部。

输入格式

第一行输入一个整数n，代表科学家的数量。
第二行输入n个整数$a_1,a_2\dots a_n$,其中$a_i$表示第i个科学家懂得的语言的编号。
第三行输入一个整数m，代表电影的数量。
第四行输入m个整数$b_1,b_2\dots b_m$,其中$b_i$表示第i部电影的语音采用的语言的编号。
第五行输入m个整数$c_1,c_2\dots c_m$,其中$c_i$表示第i部电影的字幕采用的语言的编号。
请注意对于同一部电影来说，$b_i\ne c_i$。
同一行内数字用空格隔开。

输出格式

输出一个整数，代表最终选择的电影的编号。
如果答案不唯一，输出任意一个均可。

数据范围

$1\le n,m\le 200000$,
$1\le a_i,b_i,c_i\le 10^9$

输入样例：####

3
2 3 2
2
3 2
2 3

输出样例：

2

题解

如果能听懂电影的语音，他就会很开心；如果能看懂字幕，他就会比较开心。

题目要我们选择观影很开心的人最多的电影，如果有多部电影符合要求，就选观影比较开心的人最多的那一部。

也就是在 听懂语音的人最多的电影 中挑出 看懂字幕的人最多的电影。

我们首先需要记录每一种语言有多少人懂，输入会给出每位观众的语言，离散化可以直接用map存每种语言懂的人数。

然后去记录每部电影 能听懂的人数 和 能看懂的人数，排序找出答案即可。

C++ 代码

#include <bits/stdc++.h>
using namespace std;

struct Node { int inx, a, b; };
const int N = 200000 + 10;
int n, m, x; Node movs[N];
unordered_map<int, int> mp; //unordered_map比map快

int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i ++)
    {
        scanf("%d", &x);
        mp[x] ++; //记录每种语言有多少个科学家懂
    }
    
    scanf("%d", &m);
    for (int i = 1; i <= m; i ++) 
    {
        scanf("%d", &x);
        movs[i].a = mp[x]; //第i部电影用的语音有多少个科学家懂
        movs[i].inx = i; //顺便记录一下当前的电影是第几部
    }
    
    for (int i = 1; i <= m; i ++)
    {
        scanf("%d", &x);
        movs[i].b = mp[x]; //第i部电影用的字幕有多少个科学家懂
    }
    
    sort(movs + 1, movs + 1 + n, 
    [](Node& A, Node& B) //根据题目进行优先级排序
    {
        if (A.a == B.a) return A.b > B.b;
        return A.a > B.a;
    });
    
    cout << movs[1].inx; //输出排序完第一个的序号
    
    return 0;
}

需要注意的是不要把mp[x]放入sort的匿名函数中，这样的时间复杂度会提高，应当预处理出movs.a和movs.b，sort的复杂度是$nlogn$，mp[x]的复杂度是$logn$，总复杂度为$nlo{g}^{2}n$。而预处理后的复杂度为$nlogn$。