最小生成树

最小生成树

无向带权图
子图
①连通所有点
②权之和要最小
kruskal 算法 贪心：局部最优->全局最优
稀疏图 权最小
做法
①将所有边按权排序
②按权从小到大加入子图
如果边的两点在子图中已连通则不加入
③边数=顶点数-1；
描述
某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。
输入描述：
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。 当N为0时，输入结束，该用例不被处理。
输出描述：
对每个测试用例，在1行里输出最小的公路总长度。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int father[1010];
void InitDisjointSet(int n){
    for(int i=0;i<n;++i){
        father[i]=i;
    }
}
int FinDisjointSet(int u){
    if(father[u]==u){
        return u;
    }else{
        return FinDisjointSet(father[u]);
    }
}
void UnionDisjointSet(int u,int v){
    int uroot= FinDisjointSet(u);
    int vroot= FinDisjointSet(v);
    father[vroot]=uroot;
}
struct Edge{
    int u;
    int v;
    int weight;
    Edge(int _u,int _v,int _weight){
        u=_u;
        v=_v;
        weight=_weight;
    }
};
bool compare(Edge lhs,Edge rhs){
    return lhs.weight< rhs.weight;
}
int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        if(n==0){
            break;
        }
        vector<Edge> edgeVec;
        InitDisjointSet(n+1);
        for(int i=0;i<(n-1)*n/2;++i){
            int u,v,weight;
            scanf("%d%d%d",&u,&v,&weight);
            //可以用类的构造函数来简化
            Edge e(u,v,weight);
            edgeVec.push_back(e);
        }
        //kruskal
        //排序
        sort(edgeVec.begin(),edgeVec.end(), compare);
        //将每一条边加入子图
        int totalWeight=0;
        int curEdgeNum=0;
        for(int i=0;i<edgeVec.size();++i){
            int u=edgeVec[i].u;
            int v=edgeVec[i].v;
            int weight=edgeVec[i].weight;
            if(FinDisjointSet(u)!= FinDisjointSet(v)){
                UnionDisjointSet(u,v);
                ++curEdgeNum;
                totalWeight+=weight;
                if(curEdgeNum==n-1){
                    break;
                }
            }

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