670C-Cinema

670C-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

离散化，用map存储会超时

#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
static const auto io_sync_off = []() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    return nullptr;
}();

const int maxn = 200005;
int N, M, cnt = 0, tot = 0;
int peo[maxn], au[maxn], su[maxn]; //每个人的语言，声音，字幕
int arr[maxn * 3], num[maxn * 3];  //所有可能的语言，离散化的结果
int sum[maxn];

void discrete() //离散化
{
    sort(arr + 1, arr + cnt + 1);
    for (int i = 1; i <= cnt; ++i)
        if (i == 1 || arr[i] != arr[i - 1]) //去重
            num[++tot] = arr[i];
}

int query(int x) //x离散后的值
{
    return lower_bound(num + 1, num + tot + 1, x) - num;
}

int main()
{
    cin >> N;
    for (int i = 1; i <= N; ++i)
    {
        cin >> peo[i];
        arr[++cnt] = peo[i];
    }
    cin >> M;
    for (int i = 1; i <= M; ++i)
    {
        cin >> au[i];
        arr[++cnt] = au[i];
    }
    for (int i = 1; i <= M; ++i)
    {
        cin >> su[i];
        arr[++cnt] = su[i];
    }

    discrete();
    for (int i = 1; i <= N; ++i)
    {
        int x = query(peo[i]);
        ++sum[x]; //统计个人语言数
    }

    int amax = -1, smax = -1, ans = 1;
    for (int i = 1; i <= M; ++i)
    {
        int a = query(au[i]); //语音
        int s = query(su[i]); //字幕

        if (amax < sum[a])
        {
            amax = sum[a];
            smax = sum[s];
            ans = i;
        }
        else if (amax == sum[a] && smax < sum[s])
        {
            amax = sum[a];
            smax = sum[s];
            ans = i;
        }
    }

    cout << ans;
    return 0;
}