时间限制 : 1.000 sec 内存限制 : 32 MB
题目描述
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
输入
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。
当N为0时输入结束。
输出
每个测试用例的输出占一行,输出全省畅通需要的最低成本。
样例输入 Copy
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
样例输出 Copy
3
0
思路:flag=1表示已建,那么代表这条边的花费=0,所以排序的时候只需把flag=1的边排在前面。kruskal算法的时候计算总花费的时候只需要算边flag=0时的花费。
#include <iostream>
#include <cstdio>
#include <map>
#include <vector>
#include <string>
#include <memory.h>
#include <set>
#include <stack>
#include <queue>
#include <unordered_map>
#include <iomanip>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn = 5010;
struct ss {
int u, v, cost, flag;
}s[maxn];
bool cmp(ss a, ss b) {
if (a.flag != b.flag)
return a.flag > b.flag;
else
return a.cost < b.cost;
}
int father[maxn];
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;
}
int kruskal(int n, int m) {
int ans = 0, snum = 0;
for (int i = 0; i <= n; i++) {
father[i] = i;
}
sort(s, s + m, cmp);
for (int i = 0; i < m; i++) {
int fau = findfather(s[i].u), fav = findfather(s[i].v);
if (fau != fav) {
father[fau] = fav;
snum++;
if (s[i].flag == 0)
ans += s[i].cost;
if (snum == n - 1)
break;
}
}
if (snum == n - 1)
return ans;
else
return -1;
}
int main() {
int n;
while (cin >> n) {
if (n == 0)
break;
for (int i = 0; i < n*(n - 1) / 2; i++) {
cin >> s[i].u >> s[i].v >> s[i].cost >> s[i].flag;
}
int ans = kruskal(n, n*(n - 1) / 2);
cout << ans << endl;
}
return 0;
}