problem
莫斯科正在举办一个大型国际会议,有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≤n,m≤200000,
1≤ai,bi,ci≤109
输入样例:
3
2 3 2
2
3 2
2 3
输出样例:
2
- 有1e9种不同的语言
- 有n个科学家,m场电影,每场电影有语音和字幕,共n+2m种语言
- 求选一部电影能让最多的科学家听懂语音(相同情况下取字母最多的)
solution
- 虽然map很香,不过离散化模板可以学习一下
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10;
int a[maxn], b[maxn], c[maxn], sum[maxn];
int cnt, arr[maxn*3];
void discrete(){
sort(arr+1,arr+cnt+1);
cnt = unique(arr+1,arr+cnt+1)-arr-1;
}
int query(int x){
return lower_bound(arr+1,arr+cnt+1,x)-arr;
}
int main(){
//input
ios::sync_with_stdio(false);
int n; cin>>n;
for(int i = 1; i <= n; i++){
cin>>a[i]; arr[++cnt] = a[i];
}
int m; cin>>m;
for(int i = 1; i <= m; i++){
cin>>b[i]; arr[++cnt] = b[i];
}
for(int i = 1; i <= m; i++){
cin>>c[i]; arr[++cnt] = c[i];
}
//solve
discrete();
int ans = 0, bt = -1, ct = -1;
for(int i = 1; i <= n; i++)sum[query(a[i])]++;//会每种语言的人数
for(int i = 1; i <= m; i++){//更新能被看懂最多的电影
int x = query(b[i]), y = query(c[i]);
if(sum[x]>bt){
ans = i; bt = sum[x]; ct = sum[y];
}else if(sum[x]==bt && sum[y]>ct){
ans = i; bt = sum[x]; ct = sum[y];
}
}
cout<<ans<<endl;
return 0;
}