【HD 1233】还是畅通工程（最小生成树 Kruskal算法模板题）

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 40086 Accepted Submission(s): 18229

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

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

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

Sample Input
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

Sample Output
3
5

HintHint
Huge input, scanf is recommended.

Source
浙大计算机研究生复试上机考试-2006年

——————————————————————————————————————————————————
小结一下用克鲁斯卡尔算法解决最小生成树问题的思路。
首先，克鲁斯卡尔算法是 从 边 的角度出发寻找最小生成树，先按照边的长短把图中所有的边进行储存和排序，然后优先加入短的边。
因为要避免加入了多余的边，所以通过并查集来确定是否属于同一连通分支（根节点位置），（如果在边加入前就已经有了相同的根节点，加入边后即会形成环）。

——————————————————————————————————————————————————

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

using namespace std;
int n;

struct Edge{
    int TODEa;
    int TODEb;
    int val;
}edge[5010]; //记录两端点 和边的权值

bool cmp( Edge a,Edge b )
{
    return a.val < b.val; //贪心的选择最短的权值边加入
}

int par[105]; 

void init(int n)
{
    for( int i=1; i<=n; i++ ){ //有些题目没有0号，从0开始会WA  
        par[i] = i;
    }
}

int find( int x ) // 并查集 查询函数
{
    if( par[x] == x )return x;
    else return par[x] = find(par[x]); 
}

bool unite( int x,int y )  //合并 ，并返回值，确定是否加入边
{
    x = find(x);
    y = find(y);
    if( x == y ) return false;
    else{
        par[y] = x;
        return true;
    } 
}

int main()
{   
    int m;
    while( ~scanf("%d",&n),n )
    {
        init(n);    
        int count =0;
        int ans = 0;
    //  memset(edge,0,sizeof(edge));
        m = n*(n-1)/2;
        for( int i=0; i<m; i++ )
        {                   scanf("%d%d%d",&edge[i].TODEa,&edge[i].TODEb,&edge[i].val);
        }
        sort(edge,edge+m,cmp);
     // printf("%d %d\n",n,m);
      for(int i=0;count<n-1;++i)//依次选择使得所有顶点连通的边
        {
            if(unite(edge[i].TODEa,edge[i].TODEb))
            {
                ans+=edge[i].val;
                ++count;
            }
        }

        printf("%d\n",ans);
    }
}