-
题目描述:
-
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
-
输入:
-
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
-
输出:
-
对每个测试用例,在1行里输出最小的公路总长度。
-
样例输入:
-
3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0
-
样例输出:
-
3 5
最小生成树,用并查集模拟Kruskal算法
#include<iostream>
#include<algorithm>
using namespace std;
struct Node { //保存节点
int x;
int y;
int cost;
};
int root[100];
Node node[5050]; //所有边集
int findroot(int a) //寻找根节点
{
if (root[a] == -1)
return a;
else {
int temp = findroot(root[a]); //压缩路径
root[a] = temp;
return temp;
}
}
int comp(const Node &n1, const Node &n2) {
return n1.cost < n2.cost;
}
int main()
{
int i, j, N, n;
while (cin >> n && n) {
for (int i=1; i<=n; ++i)
root[i] = -1; //初始n个节点各为一集合
N = n*(n-1)/2;
for (i=1; i<=N; i++)
cin >> node[i].x >> node[i].y >> node[i].cost;
sort (node, node+N, comp); //边权小的在前先选
int sum = 0, a , b;
for(i=1; i<=N; ++i) {
a = findroot(node[i].x);
b = findroot(node[i].y);
if (a != b) { //合并集合
root[a] = b;
sum += node[i].cost;
}//if
}//for
cout << sum << endl;
} //while
return 0;
}