题目

莫斯科正在举办一个大型国际会议，有 $n$ 个来自不同国家的科学家参会。

每个科学家都只懂得一种语言。
为了方便起见，我们把世界上的所有语言用 1 到 109 之间的整数编号。

在会议结束后，所有的科学家决定一起去看场电影放松一下。

他们去的电影院里一共有 $m$ 部电影正在上映，每部电影的语音和字幕都采用不同的语言。

对于观影的科学家来说，如果能听懂电影的语音，他就会很开心；如果能看懂字幕，他就会比较开心；如果全都不懂，他就会不开心。

现在科学家们决定大家看同一场电影。

请你帮忙选择一部电影，可以让观影很开心的人最多。

如果有多部电影满足条件，则在这些电影中挑选观影比较开心的人最多的那一部。

输入格式

第一行输入一个整数 $n$，代表科学家的数量。

第二行输入 $n$ 个整数 $a_1,a_2\dots a_n$，其中 $a_i$ 表示第 $i$ 个科学家懂得的语言的编号。

第三行输入一个整数 $m$，代表电影的数量。

第四行输入 $m$ 个整数 $b_1,b_2\dots b_m$，其中 $b_i$ 表示第 $i$ 部电影的语音采用的语言的编号。

第五行输入 $m$ 个整数 $c_1,c_2\dots c_m$，其中 $c_i$ 表示第 $i$ 部电影的字幕采用的语言的编号。

请注意对于同一部电影来说，$b_i\ne c_i$。

同一行内数字用空格隔开。

输出格式

输出一个整数，代表最终选择的电影的编号。电影编号 1∼$m$。

如果答案不唯一，输出任意一个均可。

数据范围

• $1\le n,m\le 200000$
• $1\le a_i,b_i,c_i\le 10^9$

输入样例

3
2 3 2
2
3 2
2 3

输出样例

2

分析

虽然语言的范围在 int 以内，但是这 $m$ 部电影与 $n$ 个人最多涉及 $2\ast m+n$ 种语言。把所有电影和人涉及的语言放入一个数组中，排序并离散化，用一个 1 ~ $2\ast m+n$ 之间的整数代替每种语言。此时就可以用数组直接统计会上述每种语言的人的数量，从而选择满足题目要求的电影。

时间复杂度为 $O((n+m)log(n+m))$。

代码

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;const int N = 2e5 + 50;
int a[N], b[N], c[N];
int cinema[N * 3];
int tot = 0;
int k = 0;
int ans[N * 3];//查询x映射为哪个1~k之间的整数
int query(int x) {return lower_bound(cinema + 1, cinema + 1 + k, x) - cinema;
}//离散化
void discrete() {//排序sort(cinema + 1, cinema + tot + 1);//去重，k为去重后的元素个数k = unique(cinema + 1, cinema + tot + 1) - (cinema + 1);
}int main() {int n, m;cin >> n;for (int i = 1; i <= n; i++) {cin >> a[i];cinema[++tot] = a[i];}cin >> m;for (int i = 1; i <= m; i++) {cin >> b[i];cinema[++tot] = b[i];}for (int i = 1; i <= m; i++) {cin >> c[i];cinema[++tot] = c[i];}//离散化discrete(); for (int i = 1; i <= n; i++) ans[query(a[i])]++; //统计每种语言会的科学家数//遍历所有电影//res 为最终结果，初始值为1，是因为如果没有符合条件的电影就随便选择一部，电影编号是从1开始的int res = 1, ans1 = 0, ans2 = 0;for (int i = 1; i <= m; i++) {int ansb = ans[query(b[i])]; //能听懂电影的语音的人数——>开心int ansc = ans[query(c[i])]; //能看懂字幕的人数--> 比较开心//当前电影比之前的电影开心的人多 或者 开心的人和之前的人一样，但是比较开心的比之前多，则选择当前电影if (ansb > ans1 || (ansb == ans1 && ansc > ans2)) { res = i;ans1 = ansb;ans2 = ansc;}}cout << res << endl;return 0;
}