继续畅通工程

继续畅通工程

描述
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。
输入描述：
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。 当N为0时输入结束。
输出描述：
每个测试用例的输出占一行，输出全省畅通需要的最低成本。

先构造一遍图 再计算一遍路径长度

#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;
    int state;
    Edge(int _u,int _v,int _weight,int _state){
        u=_u;
        v=_v;
        weight=_weight;
        state=_state;
    }
};
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,state;
            scanf("%d%d%d%d",&u,&v,&weight,&state);
            //可以用类的构造函数来简化
            Edge e(u,v,weight,state);
            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;
            int state=edgeVec[i].state;
            if(state==1){
                UnionDisjointSet(u,v);
            }
        }//已经构建了一遍图了
        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;
                edgeVec[i].state=1;
                if(curEdgeNum==n-1){
                    break;
                }
            }
        }printf("%d\n",totalWeight);
    }
}