题目描述:
省*“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
输入:
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1 < N < 100 1< N < 100 1<N<100);随后的 N ( N − 1 ) 2 \frac{N(N-1)}{2} 2N(N−1) 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。
当N为0时输入结束。
输出:
每个测试用例的输出占一行,输出全省畅通需要的最低成本。
样例输入:
4
1 2 1 1
1 3 6 0
1 4 2 1
2 3 3 0
2 4 5 0
3 4 4 0
3
1 2 1 1
2 3 2 1
1 3 1 0
0
样例输出:
3
0
实现代码:
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxv = 101;
const int maxe = 5000;
struct edge {
int u, v;
int cost;
int flag;
} E[maxe];
bool cmp(edge a, edge b) {
return a.cost < b.cost;
}
//并查集
int father[maxv];
int findFather(int x) {
int a = x;
while(x != father[x]) {
x = father[x];
}
while(a != father[a]) {
int z = a;
a = father[a];
father[z] = x;
}
return x;
}
//克鲁斯卡尔算法,n表示结点数,m表示边数
int kruskal(int n, int m, int numEdge) {
int ans = 0;
sort(E, E + m, cmp);
for(int i = 0; i < m; i++) {
if(E[i].flag == 0) {
int faU = findFather(E[i].u);
int faV = findFather(E[i].v);
if(faU != faV) {
father[faU] = faV;
numEdge++;
ans += E[i].cost;
if(numEdge == n - 1) {
break;
}
}
}
}
if(numEdge != n - 1) {
return -1;
} else {
return ans;
}
}
int main() {
int n;
while(scanf("%d", &n) != EOF) {
if(n == 0) {
break;
}
int m = n * (n - 1) / 2;
int numEdge = 0;
for(int i = 1; i <= n; i++) {
father[i] = i;
}
for(int i = 0; i < m; i++) {
scanf("%d%d%d%d", &E[i].u, &E[i].v, &E[i].cost, &E[i].flag);
int faU = findFather(E[i].u);
int faV = findFather(E[i].v);
if(faU != faV && E[i].flag == 1) {
father[faU] = faV;
numEdge++;
}
}
if(numEdge == n - 1) {
printf("0\n");
} else {
printf("%d\n", kruskal(n, m, numEdge));
}
}
return 0;
}