hdu - 1233 - 还是畅通工程

题意：N个村庄（N<100），给出每两个村庄之间的距离，求找这N个村庄连通的最短距离和。

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

——>>这是用来试试最小生成树的Kruskal与Prim算法的增强信心的好题呀……我用Kruskal

#include <iostream>
#include <queue>

using namespace std;

const int maxn = 100 + 10;      //N ( < 100 )
struct node     //结点数据类型
{
    int v1;     //其中一个村庄
    int v2;     //另一个村庄
    int dis;        //两个村庄之间的距离
    bool operator < (const node &v) const       //为优先队列重载运算符
    {
        return dis > v.dis;
    }
};
int fa[maxn], sum;      //并查集用的fa数组与最终目标sum

int find_set(int x)     //并查集查找
{
    if(x == fa[x]) return x;
    else
    {
        fa[x] = find_set(fa[x]);
        return fa[x];
    }
}
bool union_set(node n)      //并查集合并
{
    int root_v1 = find_set(n.v1);
    int root_v2 = find_set(n.v2);
    if(root_v1 == root_v2) return 0;
    else
    {
        fa[root_v2] = root_v1;
        sum += n.dis;
        return 1;
    }
}
int main()
{
    int N, i;
    node no;
    while(cin>>N)
    {
        if(!N) return 0;
        for(i = 1; i <= N; i++) fa[i] = i;      //初始化
        priority_queue<node> pq;        //为Kruskal算法用的优先队列
        int bor = N*(N-1)/2;
        for(i = 0; i < bor; i++)
        {
            cin>>no.v1>>no.v2>>no.dis;
            pq.push(no);
        }
        sum = 0;
        while(!pq.empty())
        {
            no = pq.top();
            pq.pop();
            union_set(no);
        }
        cout<<sum<<endl;
    }
    return 0;
}