HDU 1233 還是暢通工程 Kruskal求最小成樹

題目鏈接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66965#problem/L

題意：就是並查集題意。某省調查鄉村交通狀況，得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通（但不一定有直接的公路相連，只要能間接通過公路可達即可），並要求鋪設的公路總長度爲最小。請計算最小的公路總長度。 

以前用的是Prime算法，但是現在卻幾乎全部忘卻了，由於記得並查集，所以，現在寫Kruskal還是比較順手的。

代碼：

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

using namespace std;
struct edge{
    int u,v,w;
    bool operator<(const edge &a)const{
        return w<a.w;
    }
}E[10010];

int N,M;
int pre[110];

int find(int x){
    int t=x;
    while(pre[t]!=t) t=pre[t];
    while(x!=t) pre[x]=t,x=pre[x];
    return t;
}

void Kruskal(){
    for(int i=0;i<=N;i++) pre[i]=i;
    int cnt=1,ans=0;
    for(int i=0;i<M;i++){
        int u=E[i].u,v=E[i].v,w=E[i].w;
        int fu=find(u),fv=find(v);
        if(fu==fv) continue;
        ans+=w;
        pre[fv]=fu;cnt++;
        if(cnt==N) break;
    }
    cout<<ans<<endl;
}


int main(){
    //freopen("D:\\in.txt","r",stdin);
    while(cin>>N && N){
        M=N*(N-1)/2;
        for(int i=0;i<M;i++){
            scanf("%d %d %d",&E[i].u,&E[i].v,&E[i].w);
        }
        sort(E,E+M);
        Kruskal();
    }
    return 0;
}