题目链接:https://codeforces.ml/contest/1315/problem/D
题意:给你n个数,以及每个数+1的代价,问你让所有的数都不相同的最小代价。
思路:首先考虑最简单的思路,每次我们让相同的数花费最小的+1,答案加上所需要的代价,这样一直处理到最后一个肯定可以出结果,但时间复杂度太大,会TLE。所以我们可以考虑用并查集去维护每个值所能达到的最大的位置,然后我们用 代价 * (最大的位置 - 当前的值)就是当前位置对答案的贡献。考虑贪心的思想,我们要按照代价从大到小去对结构体进行排序。具体代码如下。
AC代码:
#include <set>
#include <map>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define LL long long
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
const int maxn = 2e5 + 7;
struct node {
LL val;
LL time;
}a[maxn];
bool cmp(node aa,node bb) {
if(aa.time == bb.time) {
return aa.val < bb.val;
} else {
return aa.time > bb.time;
}
}
map<LL,LL>mp;
LL Find(int x) {
if(mp[x] == 0) return x;
else return mp[x] = Find(mp[x]);
}
void Merge(LL x,LL y) {
LL t1 = Find(x);
LL t2 = Find(y);
if(t1 != t2) {
mp[t1] = t2;
}
}
int main() {
int n;
while(~scanf("%d",&n)) {
mp.clear();
for(int i = 1 ; i <= n ; i++) {
scanf("%lld",&a[i].val);
}
for(int i = 1 ; i <= n ; i++) {
scanf("%lld",&a[i].time);
}
sort(a+1,a+n+1,cmp);
// for(int i = 1 ; i <= n ; i++) {
// cout << "i = " << i << " " << a[i].val << " " << a[i].time << endl;
// }
LL ans = 0;
for(int i = 1 ; i <= n ; i++) {
LL res = Find(a[i].val);
// cout << "res = " << res << endl;
Merge(res,res+1);
ans += a[i].time * (res - a[i].val);
}
printf("%lld\n",ans);
}
return 0;
}