pat06-图6. 公路村村通（30）

06-图6. 公路村村通（30）

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

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

输入格式说明：

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

输出格式说明：

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

样例输入与输出：

序号	输入	输出
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
2	3 1 2 3 2	-1
3	5 4 1 2 1 2 3 2 3 1 3 4 5 4	-1

---

提交代码

边是点的线性关系，因此用Kruskal算法。

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<queue>
#include<vector>
#include<cmath>
#include<string>
using namespace std;
int f[1005];
struct node{
    int u,v,cost;
};
bool cmp(node a,node b){
    return a.cost<b.cost;
}
int findfa(int a){//返回父节点
    if(a!=f[a]){
        f[a]=findfa(f[a]);
    }
    return f[a];//注意，这里返回f[a]！！
}
int main(){
    //freopen("D:\INPUT.txt","r",stdin);
    int n,m;
    scanf("%d %d",&n,&m);
    int i;
    node *load=new node[m];
    for(i=0;i<=n;i++){
        f[i]=i;
    }
    for(i=0;i<m;i++){
        scanf("%d %d %d",&load[i].u,&load[i].v,&load[i].cost);
    }
    sort(load,load+m,cmp);
    int num=1,cur=0,mincost=0;

    //cout<<num<<endl;

    /*for(i=0;i<m;i++){
        cout<<load[i].cost<<endl;
    }*/

    while(num<n&&cur<m){
        int ufa=findfa(load[cur].u);
        int vfa=findfa(load[cur].v);
        if(ufa>vfa){
            f[vfa]=ufa;
            mincost+=load[cur].cost;
            num++;
        }
        else{
            if(vfa>ufa){
                f[ufa]=vfa;
                mincost+=load[cur].cost;
                num++;
            }
        }
        cur++;
    }

    if(num==n){
        printf("%d
",mincost);
    }
    else{
        printf("-1
");
    }
    delete []load;
    return 0;
}