7-3 公路村村通(最小生成树K算法)

7-3 公路村村通(30 分)
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。

输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。

输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。

输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
输出样例:
12

#include<bits/stdc++.h>
using namespace std;
int f[10000];
struct edge{
    int u,v;//顶点u,v;
    long long cost;
}edgeOne;
edge es[10000];
bool cmp(edge e1,edge e2){
    return e1.cost<e2.cost;
}

int V,E;//村庄数和道路数
int fun(int a){
    if(f[a]==a){
        return f[a];
    }
    f[a]=fun(f[a]);
    return f[a];
}
void group(int x,int y){
    int tx=fun(x);
    int ty=fun(y);
    if(ty!=tx){
        f[ty]=tx;
    }
    return;
}
void init(){
    for(int i=1;i<=V ;i++){
        f[i]=i;
    }
}
bool same(int x,int y){
    return fun(x)==fun(y);
}
long long kruskal(){
    long long res=0;
    sort(es+1,es+E+1,cmp);
    for(int i=1;i<=E;i++){
        if(!same(es[i].u,es[i].v)){
            group(es[i].u,es[i].v);
            res+=es[i].cost;
        }
    }
    return res;
}
int main(){
    long long res=0;
    //while(scanf("%d%d",&V,&E)==2){
    scanf("%d%d",&V,&E);
    //if(V==0){
     //   break;}
    init();
    for(int i=1;i<=E;i++){
        scanf("%d%d%lld",&es[i].u,&es[i].v,&es[i].cost);
    }
    res=kruskal();
    for(int j=1;j<=V;j++){
            if(!same(j,1)){
                res=-1;
          }
    }
     if(res==-1){
        printf("-1");
     }else{
        printf("%lld",res);
     }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_38505045/article/details/80542968