莫斯科正在举办一个大型国际会议,有 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∼m。
如果答案不唯一,输出任意一个均可。
数据范围
1≤n,m≤200000,
1≤ai,bi,ci≤109
输入样例:
3
2 3 2
2
3 2
2 3
输出样例:
2
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 2e5 + 10;
int a[N], b[N], c[N], lang[N * 3], uni[N * 3];
int ans[N * 3];
int n, m;
int cnt, k;
int find(int x)
{
return lower_bound(uni + 1, uni + k + 1, x) - uni;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n;
for(int i = 1; i <= n; i ++ ) //读a:科学家都会什么语言
{
cin >> a[i];
lang[ ++ cnt] = a[i];
}
cin >> m;
for(int i = 1; i <= m; i ++ ) //读b:电影语音是什么
{
cin >> b[i];
lang[ ++ cnt] = b[i];
}
for(int i = 1; i <= m; i ++ ) //读c:电影字幕是什么
{
cin >> c[i];
lang[ ++ cnt] = c[i];
}
sort(lang + 1, lang + cnt + 1); //排序
for(int i = 1; i <= cnt; i ++ ) //去重后放入uni
{
if(i == 1 || lang[i] != lang[i - 1])
uni[ ++ k] = lang[i];
}
for(int i = 1; i <= n; i ++ ) ans[find(a[i])] ++ ; //用uni的下标记录语言,有几个人会
int res1 = 1; //最终用哪个
int res2 = 0, res3 = 0; //当前这个语言有几个人会,几个人能看懂字幕
for(int i = 1; i <= m; i ++ )
{
int x = ans[find(b[i])];
int y = ans[find(c[i])];
if(x > res2 || (x == res2 && y > res3))
{
res1 = i;
res2 = x;
res3 = y;
}
}
cout << res1 << endl;
return 0;
}