暑期集训20190725 胜地不常(paradise)

【题目描述】 给定两个长度为n的非负整数数组a,b，

【输入数据】 第一行一个整数n。 第二行n个整数a1~an。 第三行n个整数b1~bn。

【输出数据】 一行一个整数表示答案。

【样例输入】 4 1 2 3 4 2 3 3 3

【样例输出】 12

【数据范围】 对于10%的数据，n<=2000。 对于另外20%的数据，ai,bi<=2010。 对于70%的数据，n<=510^4，ai,bi<=524288，∑ai,∑bi<=10^6。 对于100%的数据，1<=n<=10^6，0<=ai,bi<=310^6，∑ai,∑bi<=10^7。

由于1+2+3+...+n=n*(n+1)/2 ∑ai <=10^7，

所以不同的ai只会有O(sqrt(sum ai) )种。

用桶排序分别记录ai和bi的出现次数，然后分别计算即可。

#include <bits/stdc++.h>
using namespace std;
int ra[3000010], rb[3000010];//bucket
int na[3000010], nb[3000010];//number
int ca[3000010], cb[3000010];//count times
int main() {
    freopen("paradise.in", "r", stdin);
    freopen("paradise.out", "w", stdout);
    int n;
    scanf("%d", &n);
    int temp;
    for(int i=0; i<n; i++) {
        scanf("%d", &temp);
        ra[temp] ++;
    }
    for(int i=0; i<n; i++) {
        scanf("%d", &temp);
        rb[temp] ++;
    }
    int cnta=0, cntb=0;
    for(int i=0; i<=3000000; i++){
        if(ra[i]){
            cnta++;
            ca[cnta] = ra[i];
            na[cnta] = i;    
        }
        if(rb[i]){
            cntb++;
            cb[cntb] = rb[i];
            nb[cntb] = i;    
        }        
    }
    long long sum=0;
    for(int i=0; i<=cnta; i++)
        for(int j=0; j<=cntb; j++){
            sum += (long long)sqrt(abs(na[i]-nb[j])) * ca[i] * cb[j];
        }
    printf("%lld", sum);
    return 0;
}