103. 电影
题目链接
103. 电影
莫斯科正在举办一个大型国际会议,有 \(n\) 个来自不同国家的科学家参会。
每个科学家都只懂得一种语言。
为了方便起见,我们把世界上的所有语言用 \(1\) 到 \(10^9\) 之间的整数编号。
在会议结束后,所有的科学家决定一起去看场电影放松一下。
他们去的电影院里一共有 \(m\) 部电影正在上映,每部电影的语音和字幕都采用不同的语言。
对于观影的科学家来说,如果能听懂电影的语音,他就会很开心;如果能看懂字幕,他就会比较开心;如果全都不懂,他就会不开心。
现在科学家们决定大家看同一场电影。
请你帮忙选择一部电影,可以让观影很开心的人最多。
如果有多部电影满足条件,则在这些电影中挑选观影比较开心的人最多的那一部。
输入格式
第一行输入一个整数 \(n\),代表科学家的数量。
第二行输入 \(n\) 个整数 \(a_1,a_2…a_n\),其中 \(a_i\) 表示第 \(i\) 个科学家懂得的语言的编号。
第三行输入一个整数 \(m\),代表电影的数量。
第四行输入 \(m\) 个整数 \(b_1,b_2…b_m\),其中 \(b_i\) 表示第 \(i\) 部电影的语音采用的语言的编号。
第五行输入 \(m\) 个整数 \(c_1,c_2…c_m\),其中 \(c_i\) 表示第 \(i\) 部电影的字幕采用的语言的编号。
请注意对于同一部电影来说,\(b_i≠c_i\)。
同一行内数字用空格隔开。
输出格式
输出一个整数,代表最终选择的电影的编号。电影编号 \(1∼m\)。
如果答案不唯一,输出任意一个均可。
数据范围
\(1≤n,m≤200000,\)
\(1≤a_i,b_i,c_i≤10^9\)
输入样例:
3
2 3 2
2
3 2
2 3
输出样例:
2
解题思路
排序
将科学家的语言和电影的语言全部放在一个数组里面,同时记录语言所对应的类型,即该语言是科学家懂的语言还是电影的语音或者是电影的字幕,排序,统计一种语言的科学家个数,再记录到对应字幕的下标处,遍历数组,统计一门科学家语言出现的次数对应的电影语音,如果次数相等,则比较对应电影字幕中的科学家语言
- 时间复杂度:\(O(nlogn)\)
代码
// Problem: 电影
// Contest: AcWing
// URL: https://www.acwing.com/problem/content/105/
// Memory Limit: 64 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)
// %%%Skyqwq
#include <bits/stdc++.h>
//#define int long long
#define help {cin.tie(NULL); cout.tie(NULL);}
#define pb push_back
#define fi first
#define se second
#define mkp make_pair
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
template <typename T> bool chkMax(T &x, T y) { return (y > x) ? x = y, 1 : 0; }
template <typename T> bool chkMin(T &x, T y) { return (y < x) ? x = y, 1 : 0; }
template <typename T> void inline read(T &x) {
int f = 1; x = 0; char s = getchar();
while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
x *= f;
}
const int N=2e5+5;
struct A
{
int x,id,index;
bool operator<(A &b)const
{
if(x!=b.x)return x<b.x;
return id<b.id;
}
}all[N*3];
int n,m,a[N],b[N],c[N],cnt,mx[N];
unordered_map<int,vector<int>> mp;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)cin>>a[i],all[++cnt]={a[i],1,i};
cin>>m;
for(int i=1;i<=m;i++)cin>>b[i],all[++cnt]={b[i],2,i};
for(int i=1;i<=m;i++)cin>>c[i],all[++cnt]={c[i],3,i};
sort(all+1,all+cnt+1);
int idx=1;
while(idx<=cnt&&all[idx].id!=1)idx++;
int lst=0,c=0;
if(idx<=cnt)lst=all[idx].x;
for(int i=idx;i<=cnt;i++)
{
if(all[i].id==1&&all[i].x==lst)c++;
if(all[i].id==2&&all[i].x==lst)mp[c].pb(all[i].index);
if(all[i].id==3&&all[i].x==lst)mx[all[i].index]=c=max(mx[all[i].index],c);
if(all[i].id==1&&all[i].x!=lst)
{
lst=all[i].x;
c=1;
}
}
int res=1,mxx=0,mmx=0;
for(auto t:mp)
{
int x=t.fi;
auto v=t.se;
if(mxx<x)
{
mxx=x;
mmx=0;
for(int i=0;i<v.size();i++)
if(mmx<=mx[v[i]])mmx=mx[v[i]],res=v[i];
}
else if(mxx==x)
{
for(int i=0;i<v.size();i++)
if(mmx<=mx[v[i]])mmx=mx[v[i]],res=v[i];
}
}
cout<<res;
return 0;
}