传送门Vijos1579
(这是个什么鬼题库??)
Description
最小生成树P.S.S在宿命的指引下找到了巫师Kismi。P.S.S希望Kismi能帮自己变成一个完全图。Kismi由于某些不可告人的原因,把这件事交给了你。 PS: 可以保证,这个最小生成树对于最后求出的完全图是唯一的。
Input
输入的第一行是一个整数n,表示生成树的节点数。
接下来有n-1行,每行有三个正整数,依次表示每条边的端点编号和边权。
(顶点的边号在1-n之间,边权
Output
一个整数ans,表示以该树为最小生成树的最小完全图的边权之和。
Sample Input
#1
3
1 2 4
2 3 7
#2
4
1 2 1
1 3 1
1 4 2
Sample Output
#1
19
#2
12
Tips
n<20000
Solution
- 反推Kruskal,重新合并点集
- 要求最小完全图,每次合并时,ans+=(点集1*点集2-1)*(边长+1)
Code
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define ll long long
using namespace std;
struct JOKER{
int x,y,v;
}e[40007];
int n,fa[20007];
ll ans,num[20007];
bool cmp(JOKER a,JOKER b){
return a.v<b.v;
}
int getfa(int x){
if (fa[x]==x) return x;
fa[x]=getfa(fa[x]);
return fa[x];
}
int main(){
scanf("%d",&n);
for (int i=1;i<n;i++) scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].v);
for (int i=1;i<=n;i++){fa[i]=i; num[i]=1;}
sort(e+1,e+n,cmp);
for (int i=1;i<n;i++){
ans+=e[i].v;
int xx=getfa(e[i].x);
int yy=getfa(e[i].y);
ans+=(num[xx]*num[yy]-1)*(e[i].v+1);
num[xx]+=num[yy];
fa[yy]=xx;
}
printf("%lld",ans);
return 0;
}