YBTOJ构造完全图-CSDN博客

本文链接：https://blog.csdn.net/Mint_hexagram/article/details/120279415

算法分析

仿照kruskal的过程，将生成树中的边按边权升序排序，将边两端点所属的点集合合并，将两个点集合中所有点向另一个点集合中连边，由于要维持最小生成树在完全图中的唯一性，则连边的边权需为e[i].w+1，
则产生的贡献为tot+=(e[i].w+1)*(cnt[e[i].u]*cnt[e[i].v]-1)+e[i].w。
维护点集合的过程用并查集维护。

Code

#include<bits/stdc++.h> 
using namespace std;
const int maxn=1e5+5;
int num,n,m,sum[maxn],fa[maxn];
struct mint
{
	int u,v,w;
}e[maxn];
int find(int x)
{
	if(x==fa[x]) return x;
	return fa[x]=find(fa[x]);	
}
bool cmp(mint a,mint b)
{
	return a.w<b.w;
}	
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i) fa[i]=i,sum[i]=1;
	for(int i=1;i<=n-1;++i) scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
 	sort(e+1,e+n,cmp);
 	long long ans=0;
	for(int i=1;i<=n-1;++i)
 	{
 		int x=find(e[i].u),y=find(e[i].v);
		if(x==y) continue;
		ans+=(long long)(sum[x]*sum[y]-1)*(e[i].w+1);
		sum[x]+=sum[y];
		fa[y]=x;
		sum[y]=0;
		ans+=e[i].w;
 	}
 	printf("%lld",ans); 
	return 0;
}

一个我遇到的问题

最后一个循环里循环到n-1会WA，循环到n才能过

今天才发现：因为我习惯性把排序写成sort(e+1,e+n+1,cmp)，于是其实把第n条不存在的边加了进来，由于没被更新，所以第n条边边权为0，而我是按照边权升序排列，于是第n条边一定会被放到第一个位置上，于是就对答案产生了影响。
不过由于这条不存在的边两端点也都是0，因此至少kruskal中加入n条边就不会对答案产生影响。