构造完全图（最小生成树）

Posted 2021-07-24 SSL_LKJ

构造完全图

解题思路

dijkstra算法

AC代码

#include<algorithm>
#include<cstdio>
using namespace std;
int n,tot,f[100005],len[100005];
long long ans;
struct node
{
	int x,y,z;
}a[100005];
bool cmp(node x,node y)
{
	return x.z<y.z;
}
int find(int x)
{
	if(f[x]==x)return x;
	return f[x]=find(f[x]);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
	 	scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
		f[i]=i;
		len[i]=1;
	}
	f[n]=n;
	len[n]=1;
	sort(a+1,a+n,cmp);
	for(int i=1;i<n;i++)//dijkstra
	{
		int xx=find(a[i].x),yy=find(a[i].y);
		if(xx!=yy)
		{
			ans+=(1ll*len[xx]*len[yy]-1)*(1ll*a[i].z+1)+1ll*a[i].z;
			f[xx]=yy;
			len[yy]+=len[xx];
			tot++;
			if(tot==n-1)break;
		}
	}
	printf("%lld",ans);
	return 0;
}

谢谢