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[];
 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=;i<=n;++i)
         cin>>a[i].first;
     for(int i=;i<=n;++i)
         cin>>a[i].second;
     sort(a+,a++n,cmp);
     long long ans=;
     for(int i=;i<=n;++i){//按照个数从小到大排序，可能会让个数小而时间多的不停++，而原本个数多时间少的却加的不多，导致答案不对，先搞时间多的代价大的才是正解
         int x=find_(a[i].first);
         if(a[i].first==x){//当前个数唯一
             int y=find_(a[i].first+);//把它合并到+1的位置
             fa[x]=y;
         }
         else{//当前已有，放到最近一个比它大且没别选中的位置
             ans+=1ll*(x-a[i].first)*a[i].second;//更新贡献
             int y=find_(x+);//合并到+1的位置（原本的位置更新后被占了）
             fa[x]=y;
         }
     }
     cout<<ans;
     return ;
 }