vijos1579
kruskal本质
显然的是经过排序后,我们枚举到的边e[i]一定是当前枚举到的生成树中,最大的一条边。
所以只要比这个大,就可以保证题目要求的最小生成树了
因为后面的边会更大,所以我们这条边能加的数量是其两个端点所在集合当前的大小-1,至于-1的原因是。。枚举到的这条边也算。
于是代码就很简单了
注意开long long 不然30分。。。
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<string>
#include<map>
#include<cstring>
#include<vector>
#define inf 1e9
#define ll long long
#define For(i,j,k) for(ll i=j;i<=k;i++)
#define Dow(i,j,k) for(ll i=k;i>=j;i--)
using namespace std;
struct edge
{
ll u,v,w;
}e[100001];
ll n,f[100001],size[100001],ans;
inline bool cmp(edge x,edge y)
{
return x.w<y.w;
}
ll get(ll x)
{
if(f[x]==x)
return x;
f[x]=get(f[x]);
size[x]=size[f[x]];
return f[x];
}
inline void merge(ll x,ll y)
{
ll tx=get(x),ty=get(y);
f[tx]=ty;
size[ty]+=size[tx];
}
void krus()
{
For(i,1,n-1)
{
ll tx=get(e[i].u),ty=get(e[i].v);
ans+=(size[tx]*size[ty]-1)*(e[i].w+1);
merge(tx,ty);
}
}
int main()
{
scanf("%d",&n);
For(i,1,n) f[i]=i,size[i]=1;
For(i,1,n-1)
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w),ans+=e[i].w;
sort(e+1,e+n,cmp);
krus();
printf("%lld\n",ans);
}