【ybtoj 3.2 最小生成树课堂过关】D.构造完全图【kruskal】

题目链接：构造完全图

分析

记录每个连通块的节点数，kruskal，合并时统计答案

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

struct lwx
{
    int x,y,d;	
}a[100001];
long long n,fa[100001],siz[100001];

int father(int k)
{
	if(fa[k]==k) return k;
	else return father(fa[k]); 
}

bool cmp(lwx l,lwx r)
{
	return l.d<r.d;
}

int main()
{
	cin>>n;
	for(int i=1;i<n;i++)
	{
		cin>>a[i].x>>a[i].y>>a[i].d;
	}
	for(int i=1;i<=n;i++)
	{
	    fa[i]=i;
	    siz[i]=1;
	}
	sort(a+1,a+n,cmp);
	long long sum=0;
	for(int i=1;i<n;i++)
	{
		int fx=father(a[i].x),fy=father(a[i].y);
		if(fx!=fy)
		{
			fa[fx]=fy;
			sum+=(siz[fx]*siz[fy]-1)*(a[i].d+1)+a[i].d;
			siz[fy]+=siz[fx];
		}
	}
	cout<<sum;
	return 0;
}