vijos1579——宿命的PSS

vijos1579
kruskal本质

显然的是经过排序后,我们枚举到的边e[i]一定是当前枚举到的生成树中,最大的一条边。

所以只要比这个大,就可以保证题目要求的最小生成树了

因为后面的边会更大,所以我们这条边能加的数量是其两个端点所在集合当前的大小-1,至于-1的原因是。。枚举到的这条边也算。

于是代码就很简单了

注意开long long 不然30分。。。

#include
#include
#include
#include
#include
#include
#include
#include
#include
#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.wif(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);
}

你可能感兴趣的:(算法,其他)