还是畅通工程【kruskal + 并查集】

题目描述

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

输入

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

输出

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

样例输入

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

样例输出

3
5

#include<bits/stdc++.h>
#define read() freopen("input.txt","r",stdin)
#define write() freopen("output.txt","w",stdout)
using namespace std;
const int maxn = 1e5+10;
int pre[maxn];
struct dot{
    int start,end,len;
}city[maxn];
int find(int x){
    return x==pre[x]?x:pre[x]=find(pre[x]);
}
void init(){
    for( int i=0; i<maxn; i++ ) pre[i]=i;
}
bool cmp(dot a,dot b){
    return a.len<b.len;
}
int main(){
    int n;
    while(cin>>n&&n){
        int sum=0;
        memset(city,0,sizeof(city));
        for( int i=0; i<n*(n-1)/2; i++ ) cin>>city[i].start>>city[i].end>>city[i].len;
        init();
        sort(city,city+n,cmp);
        for( int i=0; i<n; i++ ){
            int a=find(city[i].start);
            int b=find(city[i].end);
            if(a!=b){
                pre[a]=b;
                sum+=city[i].len;
            }
        }
        cout<<sum<<'\n';
    }  
    return 0;
}