Vijos1579 宿命的PSS

题目链接
背景

P.S.S:“我来自哪里？”
WH:“你来自一个图。”
P.S.S:“我是谁？”
WH:“你是最小生成树。”
P.S.S:“我又要到哪里去？”
WH:“你要成为一个最小完全图（边权之和最小的完全图）。”
P.S.S:“为……为什么啊？”
WH:“这是你的宿命！因为你无聊！！！P.S.S！”
描述

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

输入格式

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

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

样例输入1

3
1 2 4
2 3 7
Copy
样例输出1

19
Copy
样例2

样例输入2

4
1 2 1
1 3 1
1 4 2
Copy
样例输出2

12
Copy
限制

每个测点1s
提示

n<20000
来源

冰尘e溶化邀请赛第四题,吴豪原创题。

我们可以发现我们在用Kruskal算法时，我们在构造一颗最小生成树时，我们总是用并查集先判断一下两个集合是否在一起（即是不是连在一棵树上了），如果不在一起，然后我们会将它们合起来（设中间连接的边（u，v）的距离是val），则这两个连起来的集合（设为A和B）相互的距离dis[A[i],B[j]]都大于val（即dis[A[i],B[j]]>val(A[i],B[j]为对于集合内任意节点）），如果dis[A[i],B[j]]<=val，那么我们可以连接A[i]和B[i]，而不一定是u,v了。
那么我们就可以来做此题了，我们在实现Kruskal时，我们用number数组记录一下并查集中一个集合中节点的个数。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,f[20010];
long long number[20010];
struct node
{
    int u,v;
    long long z;
}a[20010];
bool cmp(node a,node b)
{
    if (a.z<b.z) return true;
    return false;
}
int get(int x)
{
    if (f[x]==x) return x;else f[x]=get(f[x]);
    return f[x];
}
int main()
{
    scanf("%d",&n);
    long long ans=0;
    for (int i=1;i<=n-1;i++) 
    {
        scanf("%d%d%lld",&a[i].u,&a[i].v,&a[i].z);
        ans+=a[i].z;
    }
    sort(a+1,a+n,cmp);
    for (int i=1;i<=n;i++) f[i]=i,number[i]=1;
    for (int i=1;i<=n-1;i++)
    {
        if (get(a[i].u)!=get(a[i].v))
        {
            long long num=number[get(a[i].u)];                   
             ans+=(a[i].z+1)*(number[get(a[i].u)]*number[get(a[i].v)]-1);
            f[get(a[i].u)]=get(a[i].v);
            number[get(a[i].v)]+=num;
        }
    }
    printf("%lld\n",ans);
    return 0;
}