图论-生成树-构造完全图

记得要将初始边的边权加入答案中。

#include<bits/stdc++.h>
#define rep(i,l,r) for(int i=(l);i<=(r);i++)
#define per(i,r,l) for(int i=(r);i>=(l);i--)
#define random(l,r) ((l)+rand()%((r)-(l)+1))
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int inf=1e9+10,N=1e6+1000;
const double eps=1e-6;
struct edge{
	int a,b,w;
	bool friend operator < (edge a,edge b){
		return a.w<b.w;
	}
}e[N];
int n,fa[N],cnt[N];
ll ans;
int getf(int v){ return fa[v]==v?v:fa[v]=getf(fa[v]);}
int main(){
	//ios::sync_with_stdio(false); cin.tie(0);
	cin>>n;
	rep(i,1,n-1) cin>>e[i].a>>e[i].b>>e[i].w,ans+=e[i].w;
	sort(e+1,e+1+n-1);
	rep(i,1,n) fa[i]=i,cnt[i]=1;
	rep(i,1,n-1){
		int a=e[i].a,b=e[i].b,faa=getf(a),fab=getf(b);
		if(faa==fab) continue;
		//printf("cnt[faa]=%d,cnt[fab]=%d\n",cnt[faa],cnt[fab]);
		ans+=(ll)(cnt[faa]*cnt[fab]-1)*(e[i].w+1);
		cnt[faa]+=cnt[fab]; cnt[fab]=0; fa[fab]=faa;
	}
	cout<<ans;
	return 0;
}```