hdoj 1233 还是畅通工程

还是畅通工程

原题链接：https://acm.hdu.edu.cn/showproblem.php?pid=1233

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最小的公路总长度。

 

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output

3
5

根据边权值排序道路，然后根据并查集来判断是不是成一个环

#include <queue> 
#include <stdio.h>

using namespace std;
const int maxn=100+1;
int father[maxn];

struct edge{
    int src, dest, cost;
    bool operator <(const edge & nxt) const {
        return cost>nxt.cost;
    }
};
void init(int n){
    for (int i=1;i<=n;i++){
        father[i]=i;
    }
}

int find(int a){
    int x=a;
    while(a!=father[a]){
        a=father[a];
    }
    while(x!=father[x]){
        int tmp=x;
        x=father[x];
        father[tmp]=a;
    }
    return a;
}

bool Union(int a , int b){
    int x1=find(a);
    int x2=find(b);
    if (x1!=x2){
        father[x1]=x2;
        return true;
    }else return false;
}

int main(){
    edge e;
    int n,m;
    while(scanf("%d",&n)!=EOF && n ) {
        priority_queue<edge> q;
        init(n);
        m=n*(n-1)/2;
        while(m--){
            scanf("%d%d%d", &e.src, &e.dest, &e.cost);
            q.push(e);
        }
        int ans=0,count=0;
        while(!q.empty()){
            e=q.top();
            q.pop();
            if(Union(e.src,e.dest)){ //并查集来判断是不是成环 
                count++;
                ans+=e.cost;
            }
            if(count==n-1) break; 
        }
        printf("%d
",ans);
    }
    return 0;
}