[最小生成树]vijos1579 宿命的PSS

最新推荐文章于 2018-03-07 09:48:33 发布

weixin_30699955

最新推荐文章于 2018-03-07 09:48:33 发布

阅读量46

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/OIerLYF/p/7228035.html

本文介绍了一种从给定的最小生成树构建最小完全图的方法，并提供了详细的算法实现过程，包括使用Kruskal算法的核心思想进行边权更新的具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目梗概

输入给出最小生成树，由最小生成树求出最小完全图（任意两点之间只有一条线段相连）。

思考

首先一个图中最小生成树，按照kruskal算法。一定是使图连通的最小边。我们设这两点的边权为W，那么与两点相连的其他的边一定是W+1。

所以构建方法是每次选出图中最短的边，其两点标记为A,B，边权为W，则与A,B连通的点边权修改为W+1。

#include <cstdio>
#include <algorithm>
typedef long long ll;

int n;
ll num[20005],fa[20005],ans; //num用来统计联通快的数量
struct node{
    ll u,v,w;
}e[20005];

bool CMP(const node &a,const node&b){
    return a.w<b.w;
}

ll find(int x){
    if(x!=fa[x]) fa[x] = find(fa[x]);
    return fa[x];
}

void merge(int x,int y,int z){
    x = find(x);
    y = find(y);
    ans += (num[x]*num[y]-1) * (z+1);   //这里是核心
    fa[y]=x;
    num[x]+=num[y];
}

void Run(){
    for(int i=1;i<=n;i++){
        fa[i]=i;
        num[i]=1;
    }
    for(int i=1;i<n;i++){
        ans+=e[i].w; //加上两点的边
        merge(e[i].u,e[i].v,e[i].w);
    }
}

void init(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%lld%lld%lld",&e[i].u,&e[i].v,&e[i].w);
    }
    std::sort(e+1,e+n,CMP);
}

int main(){
    init();    
    Run();
    printf("%lld\n",ans);
    return 0;
}