08-图7 公路村村通

08-图7 公路村村通

分数 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

代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
C++ (g++)

思路：

Kruskal算法形成最小生成树。将已经建好的路放入一个集合中，按照费用（权值）从小到大的顺序对数组进行排序，如果两个镇子连通（即同一个父节点）的话，总数++，总花费加上该路的花费。
如果能联通，那么输出最小花费，否则输出-1。

AC代码：

#include<bits/stdc++.h>
using namespace std;

int m,n;

typedef struct{
    int u;
    int v;
    int w;
}Node;

Node node[3001];
int f[1001];

void init(){
    for(int i=0;i<=n;i++)
        f[i]=i;
}

int find(int x){
    if(x==f[x])
        return x;
    else
        return f[x]=find(f[x]);
}

int merge(int x,int y){
    int a=find(x);
    int b=find(y);
    if(a!=b){
        f[b]=a;
        return 1;
    }
    return 0;
}

bool cmp(Node n1,Node n2){
    return n1.w<n2.w;
}

int main(){
    cin>>n>>m;
    init();
    for(int i=1;i<=m;i++){
        cin>>node[i].u>>node[i].v>>node[i].w;
    }
    sort(node+1,node+m+1,cmp);
    int cnt=0,sum=0;
    for(int i=1;i<=m;i++){
        if(merge(node[i].u,node[i].v)){
            cnt++;
            sum+=node[i].w;
        }
        if(cnt==n-1){
            cout<<sum;
            return 0;
        }
    }
    cout<<-1;
    return 0;
}