题目地址:
https://www.acwing.com/problem/content/description/105/
莫斯科正在举办一个大型国际会议,有 n n n个来自不同国家的科学家参会。每个科学家都只懂得一种语言。为了方便起见,我们把世界上的所有语言用 1 1 1到 1 0 9 10^9 109之间的整数编号。在会议结束后,所有的科学家决定一起去看场电影放松一下。他们去的电影院里一共有 m m m部电影正在上映,每部电影的语音和字幕都采用不同的语言。对于观影的科学家来说,如果能听懂电影的语音,他就会很开心;如果能看懂字幕,他就会比较开心;如果全都不懂,他就会不开心。现在科学家们决定大家看同一场电影。请你帮忙选择一部电影,可以让观影很开心的人最多。如果有多部电影满足条件,则在这些电影中挑选观影比较开心的人最多的那一部。
输入格式:
第一行输入一个整数
n
n
n,代表科学家的数量。
第二行输入
n
n
n个整数
a
1
,
a
2
…
a
n
a_1,a_2…a_n
a1,a2…an,其中
a
i
a_i
ai表示第
i
i
i个科学家懂得的语言的编号。
第三行输入一个整数
m
m
m,代表电影的数量。
第四行输入
m
m
m个整数
b
1
,
b
2
…
b
m
b_1,b_2…b_m
b1,b2…bm,其中
b
i
b_i
bi表示第
i
i
i部电影的语音采用的语言的编号。
第五行输入
m
m
m个整数
c
1
,
c
2
…
c
m
c_1,c_2…c_m
c1,c2…cm,其中
c
i
c_i
ci表示第
i
i
i部电影的字幕采用的语言的编号。
请注意对于同一部电影来说,
b
i
≠
c
i
b_i≠c_i
bi=ci。
同一行内数字用空格隔开。
输出格式:
输出一个整数,代表最终选择的电影的编号。电影编号
1
∼
m
1∼m
1∼m。
如果答案不唯一,输出任意一个均可。
数据范围:
1
≤
n
,
m
≤
200000
1≤n,m≤200000
1≤n,m≤200000
1
≤
a
i
,
b
i
,
c
i
≤
1
0
9
1≤a_i,b_i,c_i≤10^9
1≤ai,bi,ci≤109
先统计每个语言有多少个科学家懂,然后遍历电影找到最优的那个即可。代码如下:
#include <iostream>
#include <unordered_map>
using namespace std;
const int N = 2e5 + 10;
int n, m;
int b[N], c[N];
int main() {
scanf("%d", &n);
unordered_map<int, int> mp;
for (int i = 1; i <= n; i++) {
int a;
scanf("%d", &a);
mp[a]++;
}
scanf("%d", &m);
for (int i = 1; i <= m; i++) scanf("%d", &b[i]);
for (int i = 1; i <= m; i++) scanf("%d", &c[i]);
int res = -1, max_yuyin = -1, max_zimu = -1;
for (int i = 1; i <= m; i++) {
int yuyin = mp[b[i]];
if (yuyin > max_yuyin)
res = i, max_yuyin = yuyin, max_zimu = mp[c[i]];
else if (yuyin == max_yuyin) {
if (mp[c[i]] > max_zimu) res = i, max_zimu = mp[c[i]];
}
}
printf("%d\n", res);
}
时空复杂度 O ( n + m ) O(n+m) O(n+m)。