算法复习 - AcWing最小生成树

346. 走廊泼水节

//考虑Kru
//假设我们在生成最小树的过程中，计算使生成的连通块成为完全图的最小边长
//枚举到某一条边的时候，假设他连接AB两个连通块
//生成新的大连通块需要AB所有点连接
//而且新加的边必须大于当前枚举的这条边
//否则生成的树就不是以前的样子了
vector<pair<int,pair<int,int>>>mp;
int f[MX],siz[MX];
int find(int x)
{
    if(x==f[x]) return x;
    return f[x]=find(f[x]);
}
void solve()
{
    int n;cin>>n;
    mp.clear();
    rpp(i,n) f[i]=i,siz[i]=1;
    rpp(i,n-1)
    {
        int x,y,w;cin>>x>>y>>w;
        mp.push_back(make_pair(w,make_pair(x,y)));
    }
    sort(all(mp));
    ll ans=0;
    for(auto edge:mp)
    {
        int x = edge.second.first,y=edge.second.second,w = edge.first;
        int fx=find(x),fy=find(y);
        if(fx==fy) continue;
        ans+=(w+1)*(siz[fx]*siz[fy]-1);
        siz[fy]+=siz[fx],f[fx]=fy;
    }
    cout<<ans<<endl;
}