B - 畅通工程 HDU - 1863 kruskal算法

problem
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。
Output
对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。
算法讲解指路

#include <iostream>
#include <vector>
#include <cstdio>
#include <cstring>
#include <math.h>
#include <algorithm>
#include <queue>
using namespace std;

typedef long long ll;
const int N = 10006;
const int inf = 0x3f3f3f3f;

struct node
{
    int u, v, w; //起点，终点，权值
}e[N];

int pre[N];

bool cmp(node a, node b)
{
    return a.w < b.w;
}

void init()
{
    for(int i = 1; i <= N; ++i)
        pre[i] = i;
}

int found(int a)
{
    int root = pre[a];
    if(root == pre[root])
        return root;
    while(pre[root] != root)
        root = pre[root];
    int tp;
    while(a != pre[a]) // 压缩路径
    {
        tp = pre[a];
        pre[a] = root;
        a = tp;
    }
    return root;
}

void unionn(int a, int b)
{
    int aa = found(a);
    int bb = found(b);
    if(aa != bb)
        pre[aa] = bb;
}

int main()
{
    int m, u, v, w, n, i;
    while(~scanf("%d%d", &n, &m))
    {
        if(!n)
            break;
        int cnt, ans;
        cnt = ans = 0;
        init();
        memset(e, 0, sizeof e);
        for(i = 1; i <= n; ++i)   //输入边的情况
            scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
        sort(e, e+n, cmp);  //按权值排序
        for(i = 1; i <= n; ++i)
        {
            if(cnt == m-1)  //所有点都已加入
                break;
            if(found(e[i].u) != found(e[i].v)) //不在一个集合
            {
                unionn(e[i].u, e[i].v);
                ans += e[i].w;
                ++cnt;
            }
        }
        if(cnt == m-1) //
            printf("%d\n", ans);
        else
            printf("?\n");
    }
    return 0;
}