题目描述
莫斯科正在举办一个大型国际会议,有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
算法
(离散化) $O((m+n)log(2*m+n))$
这题题意看着好迷啊...最后让输出的编号是指bi、ci的下标i...
既然这题是练习离散化的就用一下离散化吧,当然用map存也可以。先进行一下离散化然后统计一下a数组中
的每个数出现的次数,然后遍历一下b数组,看看b数组中在a数组里面存在的并且出现次数最多的数在b数组
里的编号是多少,如果有多个数的话再看c数组中在a数组里面存在而且出现次数最多的数在c数组里的编号是
多少就行了…
参考文献
《算法竞赛进阶指南》
C++ 代码
//https://www.cnblogs.com/shuitiangong/
#include<set>
#include<map>
#include<list>
#include<stack>
#include<queue>
#include<cmath>
#include<cstdio>
#include<cctype>
#include<string>
#include<vector>
#include<climits>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define endl '\n'
#define rtl rt<<1
#define rtr rt<<1|1
#define lson rt<<1, l, mid
#define rson rt<<1|1, mid+1, r
#define zero(a) memset(a, 0, sizeof(a))
#define INF(a) memset(a, 0x3f, sizeof(a))
#define IOS ios::sync_with_stdio(false)
#define _test printf("==================================================\n")
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
typedef pair<ll, ll> P2;
const double pi = acos(-1.0);
const double eps = 1e-7;
const ll MOD = 1000000009;
const int INF = 0x3f3f3f3f;
template<typename T> void read(T &x){
x = 0;char ch = getchar();ll f = 1;
while(!isdigit(ch)){if(ch == '-')f*=-1;ch=getchar();}
while(isdigit(ch)){x = x*10+ch-48;ch=getchar();}x*=f;
}
const int maxn = 2e5+10;
int a[maxn], b[maxn], c[maxn], d[maxn], cnt[maxn];
int bnsh(int l, int r, int x) {
while(l<r) {
int mid = (l+r+1)>>1;
if (x>=a[mid]) l = mid;
else r = mid-1;
}
return a[l] == x ? l : -1;
//我们二分查找在a数组是否存在这个数
//如果不存在就是-1存在就是它的下标,这样来进行离散化
}
int main(void) {
int n, m;
scanf("%d", &n);
for (int i = 0; i<n; ++i)
scanf("%d", &a[i]);
sort(a, a+n);
int len = unique(a, a+n)-a-1; //去重
scanf("%d", &m);
for (int i = 0; i<m; ++i)
scanf("%d", &b[i]);
for (int i = 0; i<m; ++i)
scanf("%d", &c[i]);
for (int i = 0; i<n; ++i) {
a[i] = bnsh(0, len, a[i]);
//对数组a离散化并统计数量
++cnt[a[i]];
}
for (int i = 0; i<m; ++i) {
//对数组b和c离散化,之所以不和上面放一起是因为n和m大小不同
b[i] = bnsh(0, len, b[i]);
c[i] = bnsh(0, len, c[i]);
}
int maxx = 0, maxx2 = 0, ans = 1;
for (int i = 0; i<m; ++i) {
if (maxx<cnt[b[i]] && b[i]!=-1) {
ans = i+1;
maxx = cnt[b[i]];
maxx2 = cnt[c[i]];
}
else if (c[i]!=-1 && b[i]!=-1 && maxx==cnt[b[i]] && maxx2<cnt[c[i]]) {
ans = i+1;
maxx2 = cnt[c[i]];
}
}
printf("%d\n", ans);
return 0;
}