记得要将初始边的边权加入答案中。
#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;
}```