[Vijos1579]宿命的PSS

传送门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;
}