Codeforces 670C C. Cinema (排序+离散化)

Moscow is hosting a major international conference, which is attended by n scientists from different countries. Each of the scientists knows exactly one language. For convenience, we enumerate all languages of the world with integers from 1 to 109.

In the evening after the conference, all n scientists decided to go to the cinema. There are m movies in the cinema they came to. Each of the movies is characterized by two distinct numbers — the index of audio language and the index of subtitles language. The scientist, who came to the movie, will be very pleased if he knows the audio language of the movie, will be almost satisfied if he knows the language of subtitles and will be not satisfied if he does not know neither one nor the other (note that the audio language and the subtitles language for each movie are always different).

Scientists decided to go together to the same movie. You have to help them choose the movie, such that the number of very pleased scientists is maximum possible. If there are several such movies, select among them one that will maximize the number of almost satisfied scientists.

Input
The first line of the input contains a positive integer n (1 ≤ n ≤ 200 000) — the number of scientists.

The second line contains n positive integers a1, a2, …, an (1 ≤ ai ≤ 109), where ai is the index of a language, which the i-th scientist knows.

The third line contains a positive integer m (1 ≤ m ≤ 200 000) — the number of movies in the cinema.

The fourth line contains m positive integers b1, b2, …, bm (1 ≤ bj ≤ 109), where bj is the index of the audio language of the j-th movie.

The fifth line contains m positive integers c1, c2, …, cm (1 ≤ cj ≤ 109), where cj is the index of subtitles language of the j-th movie.

It is guaranteed that audio languages and subtitles language are different for each movie, that is bj ≠ cj.

Output
Print the single integer — the index of a movie to which scientists should go. After viewing this movie the number of very pleased scientists should be maximum possible. If in the cinema there are several such movies, you need to choose among them one, after viewing which there will be the maximum possible number of almost satisfied scientists.

If there are several possible answers print any of them.

Examples
input
3
2 3 2
2
3 2
2 3
output
2
input
6
6 3 1 1 3 7
5
1 2 3 4 5
2 3 4 5 1
output
1
Note
In the first sample, scientists must go to the movie with the index 2, as in such case the 1-th and the 3-rd scientists will be very pleased and the 2-nd scientist will be almost satisfied.

In the second test case scientists can go either to the movie with the index 1 or the index 3. After viewing any of these movies exactly two scientists will be very pleased and all the others will be not satisfied.

主要考察了离散化知识点
数据处理过程使用了sort排序和stl中二分查找lower_bound
补充

在有序数组int数组(元素存放在下标1~n)中查找大于等于x的最小整数的下标：
int i=lower_bound(a+1,a+n+1,x)-a;
在有序vector<int>中查找小于等于x的最大整数(假设一定存在)
int y=*--upper_bound(a.begin(),a.end(),x);

#include <cstdio>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=2e5+5;
int n,m;
int a[maxn],b[maxn],c[maxn];
int sum[maxn*3];
int cnt,mm;
int arr[maxn*3],num[maxn*3];
void discrete(){//离散化
    sort(arr+1,arr+cnt+1);
    for(int i=1;i<=cnt;i++){
        if(i==1||arr[i]!=arr[i-1])
            num[++mm]=arr[i];
    }
}
int query(int x){//二分查找x的位置
    return lower_bound(num+1,num+mm+1,x)-num;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){//将所有电影和人涉及的语言放进一个数组，排序并离散化
        scanf("%d",&a[i]);
        arr[++cnt]=a[i];
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d",&b[i]);
        arr[++cnt]=b[i];
    }
    for(int i=1;i<=m;i++){
        scanf("%d",&c[i]);
        arr[++cnt]=c[i];
    }
    discrete();//离散化
    for(int i=1;i<=n;i++){
        int id=query(a[i]);//统计每种语言的人的数量
        ++sum[id];
    }
    int bmax=-1,cmax=-1,ans=0;
    for(int i=1;i<=m;i++){//选择满足题目要求的电影
        int x=query(b[i]);
        int y=query(c[i]);
        if(sum[x]>bmax){//优先考虑让很高兴的人最多
            bmax=sum[x],cmax=sum[y];
            ans=i;
        }else {
            if(sum[x]==bmax&&sum[y]>cmax){//如果答案不唯一、则在此前提下再让比较高兴的人最多
                bmax=sum[x],cmax=sum[y];
                ans=i;
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}

来源：算法竞赛进阶指南