继续畅通工程
描述
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
输入描述:
测试输入包含若干测试用例。每个测试用例的第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);
}
}