HDU 1233还是畅通工程（最小生成树）

HDU 1233还是畅通工程（简单最小生成树）

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

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

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

刚学习完并查集合最小生成树的相关算法，所以采用kruskal算法构建最小生成树，通过最小堆存储所有边的权值，依次取出当前权值最小的边，当该条边不属于已经有的连通分支的时候，将该边加入已有的连同分支。

AC代码：

#include <cstdio>
#include <queue>
#include <vector>

using namespace std;

typedef struct Edge
{
    int a, b;
    int weight;
    bool operator < (Edge e) const       
    {
        return weight < e.weight;
    }
    bool operator > (Edge e) const
    {
        return weight > e.weight;
    }
}Edge;

int pre[105];

int Find(int x)
{
    int r = x;
    while(pre[r] != r)
    {
        r = pre[r];
    }

    int i = x, j;
    while(i != r)
    {
        j = pre[i];
        pre[i] = r;
        i = j;
    }
    return r;
}

void join(int x, int y)
{
    int fx = Find(x), fy = Find(y);
    if(fx != fy)
    {
        pre[fx] = fy;
    }
}

int main()
{
    int n;
    while(~scanf("%d", &n) && n != 0)
    {
        Edge e[n * (n - 1) / 2];
        int x, y, w;
        priority_queue<Edge, vector<Edge>, greater<Edge> > q;
        for(int i = 0; i < n * (n - 1) / 2; i++)
        {
            scanf("%d %d %d", &x, &y, &w);
            e[i].a = x;
            e[i].b = y;
            e[i].weight = w;
            q.push(e[i]);
            pre[x] = x;
            pre[y] = y;
        }

        int sum = 0;

        for(int i = 0; i < n * (n - 1) / 2; i++)
        {
            if(Find(q.top().a) != Find(q.top().b))
            {
                join(q.top().a, q.top().b);
                sum += q.top().weight;
                q.pop();
            }
            else
            {
                q.pop();
            }
        }

        printf("%d\n", sum);
    }
    return 0;
}