九度 题目1017：还是畅通工程

题目来源：http://ac.jobdu.com/problem.php?pid=1017

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：3012

解决：1506

题目描述：

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

输入：

    测试输入包含若干测试用例。每个测试用例的第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

来源：

2006年浙江大学计算机及软件工程研究生机试真题

Kruskal是边为主导，首先要把边按升序排列，每次选取的都是当前权值最小的边，如果该边所关联的两个顶点在一个集合里，那么就不做处理，否则，就要把这两顶点所在的顶点集合融合到一个集合里（参考并查集，网上有大把的资料），也就是相当于把该边加入到最小生成树中，参考： http://blog.csdn.net/hearthougan/article/details/16961485

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

using namespace std;

const int MAXN = 110;

struct Node
{
    int u, v;
    int iWeight;
    Node()
    {
        u = v = iWeight = 0;
    }
};

int n, m;
int Parent[MAXN], iCount[MAXN];
Node Edge[MAXN];

void Inite()
{
    int i;
    for(i = 0; i <= n; ++i)
    {
        Parent[i] = i;
        iCount[i] = 1;
    }
}

int Find(int u)
{
    int s = u;
    for(; s != Parent[s]; s = Parent[s]) ;//找出根节点，因为只有根节点的父节点等于他本身
    while(s != u)
    {
        int tmp = Parent[u];
        Parent[u] = s;
        u = tmp;
    }
    return s;
}

void Union(int u, int v)
{
    int r1 = Find(u);
    int r2 = Find(v);
    if(iCount[r1] > iCount[r2])
    {
        iCount[r1] += iCount[r2];
        Parent[r2] = r1;
    }
    else
    {
        iCount[r2] += iCount[r1];
        Parent[r1] = r2;
    }
}


int Kruskal()
{
    int iSum = 0, iNum = 0;
    int u, v, i;
    Inite();
    for(i = 0; i < m; ++i)
    {
        u = Edge[i].u;
        v = Edge[i].v;
        if(Find(u) != Find(v))
        {
            iSum += Edge[i].iWeight;
            iNum++;
            Union(u, v);
        }
        if(iNum > n-1)
            break;
    }
    return iSum;
}

int CMP(Node a, Node b)
{
    return a.iWeight < b.iWeight;
}
int main()
{
    int i, u, v, w;
    while(~scanf("%d", &n) && n)
    {
        m = n*(n-1)/2;
        for(i = 0; i < m; ++i)
        {
            scanf("%d %d %d", &u, &v, &w);
            Edge[i].u = u, Edge[i].v = v, Edge[i].iWeight = w;
        }
        sort(Edge, Edge+m, CMP);
        int iSum = Kruskal();
        printf("%d\n", iSum);
    }
    return 0;
}