题目描述
莫斯科正在举办一个大型国际会议,有n个来自不同国家的科学家参会。
每个科学家都只懂得一种语言。
为了方便起见,我们把世界上的所有语言用$1$到$10^9$之间的整数编号。
在会议结束后,所有的科学家决定一起去看场电影放松一下。
他们去的电影院里一共有m部电影正在上映,每部电影的语音和字幕都采用不同的语言。
对于观影的科学家来说,如果能听懂电影的语音,他就会很开心;如果能看懂字幕,他就会比较开心;如果全都不懂,他就会不开心。
现在科学家们决定大家看同一场电影。
请你帮忙选择一部电影,可以让观影很开心的人最多。
如果有多部电影满足条件,则在这些电影中挑选观影比较开心的人最多的那一部。
输入格式
第一行输入一个整数n,代表科学家的数量。
第二行输入n个整数$a_1,a_2 \ldots a_n$,其中$a_i$表示第i个科学家懂得的语言的编号。
第三行输入一个整数m,代表电影的数量。
第四行输入m个整数$b_1,b_2 \ldots b_m$,其中$b_i$表示第i部电影的语音采用的语言的编号。
第五行输入m个整数$c_1,c_2 \ldots c_m$,其中$c_i$表示第i部电影的字幕采用的语言的编号。
请注意对于同一部电影来说,$b_i \neq 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 <= n; i ++)
{
scanf("%d", &x);
movs[i].a = mp[x]; //第i部电影用的语音有多少个科学家懂
movs[i].inx = i; //顺便记录一下当前的电影是第几部
}
for (int i = 1; i <= n; 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$,总复杂度为$nlog^2n$。而预处理后的复杂度为$nlogn$。