构造完全图--并查集+Kruskal思想

Loj 10067

题目分析：

• 要使完全图G的最小生成树为唯一T，我们按照Kruskal思想将边从小到大排序，然后依次加
• 一条边连接两边的集合，由于完全图每两个点之间都有边，则对于$x和y$这两个集合需要的边数为$cnt[x]\ast cnt[y]$，由于我们要找最小完全图，并且图G的最小生成树为T，那么：则需要一条边必须为树T的边，则新建边的数量为$cnt[x]\ast cnt[y]-1$，而又要使图G最小，则新建边的权值为$e[i].w+1$,那么$ans+=(cnt[x]\ast cnt[y]-1)\ast (e[i].w+1)$,再将两个合并起来，重复此过程

Code：

#include <bits/stdc++.h>
using namespace std;
#define maxn 110000

int cnt[maxn],n,f[maxn];
long long ans=0;
struct edge {
	int u,v,w;
}e[maxn];

inline void init_() {
	freopen("a.txt","r",stdin);
}

inline int read_() {
	int x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9') {
		if(c=='-') f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9') {
		x=(x<<3)+(x<<1)+c-'0';
		c=getchar();
	}
	return x*f;
}

int find_(int x) {
	if(f[x]==x) return x;
	else return f[x]=find_(f[x]);
}

inline void clean_() {
	for(int i=0;i<=n+1;++i) {
		f[i]=i;cnt[i]=1;
	}
}

bool cmp_(edge aa,edge bb) {
	return aa.w<bb.w;
}

void readda_() {
	n=read_();
	clean_();
	for(int i=1;i<n;++i) {
		e[i].u=read_();e[i].v=read_();e[i].w=read_();
	}
	sort(e+1,e+n,cmp_);
	for(int i=1;i<n;++i) {
		int x=find_(e[i].u),y=find_(e[i].v);
	//	if(x==y) continue;
		ans+=(long long ) ( (long long ) cnt[x]*cnt[y]-1 ) * ( (long long ) e[i].w+1 );
		ans+=(long long) e[i].w; 
		f[x]=y;
		cnt[y]+=cnt[x];
	}	
	printf("%lld",ans);
}

int main() {
	init_();
	readda_();
	return 0;
}