题目描述
莫斯科正在举办一个大型国际会议,有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
思路1:
注意:对于电影来说,满意,一般和不满意,权重是不一样的,所以必须重写比较函数
使用unordered_map存放懂每种语言的科学家的人数;
使用pair<pair<b,c>,i>
存放i号电影,能听懂的科学家的人数b,能看懂字幕的科学家人数c;
最后进行排序,输出最大值
C++ 代码
#include <iostream>
#include <map>
#include <algorithm>
#include <unordered_map>
using namespace std;
const int N=200010;
typedef pair<pair<int,int>,int> PIII;
int b[N],c[N];
PIII mov[N];
unordered_map<int,int> res;
bool compare(PIII p1,PIII p2){
if(p1.first.first == p2.first.first) return p1.first.second>p2.first.second;
return p1.first.first>p2.first.first;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
int a,n,m;
cin>>n;
for(int i=0;i<n;i++) {
cin>>a;
res[a]++;
}
cin>>m;
for(int i=1;i<=m;i++) cin>>b[i];
for(int i=1;i<=m;i++) cin>>c[i];
for(int i=1;i<=m;i++) mov[i]=make_pair(make_pair(res[b[i]],res[c[i]]),i);
sort(mov+1,mov+m+1,compare);
cout<<mov[1].second;
return 0;
}
思路2
注意:对于电影来说,满意,一般和不满意,权重是不一样的,所以可以适当提高a[b[i]]即能听懂该电影的科学家人数,比如变为原来的10倍
C++ 代码
#include <iostream>
#include <map>
#include <algorithm>
#include <unordered_map>
using namespace std;
const int N=200010;
typedef pair<int,int> PII;
int b[N],c[N];
PII mov[N];
unordered_map<int,int> res;
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
int a,n,m;
cin>>n;
for(int i=0;i<n;i++) {
cin>>a;
res[a]++;
}
cin>>m;
for(int i=1;i<=m;i++) cin>>b[i];
for(int i=1;i<=m;i++) cin>>c[i];
for(int i=1;i<=m;i++) mov[i]=make_pair(10*res[b[i]]+res[c[i]],i);
sort(mov+1,mov+m+1);
cout<<mov[m].second;
return 0;
}
为什么用PIII排序就不会超时了QAQ…难顶
6