宿命的PSS

宿命的PSS
题目描述
最小生成树P.S.S在宿命的指引下找到了巫师Kismi。P.S.S希望Kismi能帮自己变成一个完全图。Kismi由于某些不可告人的原因，把这件事交给了你。 PS: 可以保证，这个最小生成树对于最后求出的完全图是唯一的。

输入
输入的第一行是一个整数n，表示生成树的节点数。 接下来有n-1行，每行有三个正整数，依次表示每条边的端点编号和边权。 （顶点的边号在1-n之间，边权< maxint）

输出
一个整数ans，表示以该树为最小生成树的最小完全图的边权之和。

样例输入
3
1 2 4
2 3 7
样例输出
19
提示
n< 20000

$$/此题根据最小生成树的概念：给了构成最小生成树的边，就说明它是那一步中(u,v)最小的，所以其它的至少大于它/$$
$$/因为是完全图，所以两个集合间必定两两连边。综上，就按照构成最小生成树的步骤来，中途加入统计需要加的和/$$

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define ll long long
struct node{
  int u,v,c;
}f[20005];
bool cmp(node x,node y){
  return x.c<y.c;
}
int fa[20005]; 
int find(int x){
  if (fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
ll num[20005];
int main(){
  int n;
    scanf("%d",&n);
    for (int i=1;i<n;++i) scanf("%d%d%d",&f[i].u,&f[i].v,&f[i].c);
  sort(f+1,f+n,cmp);
    for (int i=1;i<=n;++i) fa[i]=i,num[i]=1;
    ll ans=0;
    for (int i=1;i<n;++i){
        ans+=f[i].c;
      int x=find(f[i].u);
        int y=find(f[i].v);
    ans+=(num[x]*num[y]-1)*(f[i].c+1);
        num[x]+=num[y];
        fa[y]=x;
    }
    printf("%lld\n",ans);
    return 0;
}