Codeforces Round #623 (Div. 1, based on VK Cup 2019-2020 - Elimination Round, Engine)A（模拟，并查集）

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
pair<int,int>a[200007];
bool cmp(pair<int,int>a,pair<int,int>b){
    if(a.second!=b.second)
        return a.second>b.second;//按照时间从大到小排序
    return a.first<b.first;
}
map<int,int>fa;
int find_(int x){
    if(!fa[x])
        return x;
    return fa[x]=find_(fa[x]);
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>a[i].first;
    for(int i=1;i<=n;++i)
        cin>>a[i].second;
    sort(a+1,a+1+n,cmp);
    long long ans=0;
    for(int i=1;i<=n;++i){//按照个数从小到大排序，可能会让个数小而时间多的不停++，而原本个数多时间少的却加的不多，导致答案不对，先搞时间多的代价大的才是正解
        int x=find_(a[i].first);
        if(a[i].first==x){//当前个数唯一
            int y=find_(a[i].first+1);//把它合并到+1的位置
            fa[x]=y;
        }
        else{//当前已有，放到最近一个比它大且没别选中的位置
            ans+=1ll*(x-a[i].first)*a[i].second;//更新贡献
            int y=find_(x+1);//合并到+1的位置（原本的位置更新后被占了）
            fa[x]=y;
        }
    }
    cout<<ans;
    return 0;
}