给定一个包含n个顶点的正权无向图,编号为0至n-1。请编写程序求出其最小支撑树,并计算其代价。
输入格式:
输入包含多组数据。每组数据第一行为2个整数n和e,均不超过200,分别表示图的顶点数和边数。接下来e行表示每条边的信息,每行为3个非负整数a、b、c,其中a和b表示该边的端点编号,c表示权值。各边并非按端点编号顺序排列。
输出格式:
对于每组数据,若存在最小支撑树则输出一个整数,为最小支撑树各边权值之和;若不存在最小支撑树,则输出“There is no minimum spanning tree.”。
输入样例:
4 5
0 1 1
0 3 1
1 3 5
1 2 1
2 3 8
4 2
0 1 1
2 3 8
输出样例:
3
There is no minimum spanning tree.
用的Kruskal算法,最小生成树问题,记得每次初始化并查集数组。只有连通图才存在最小生成树。恰选中n-1条边即可
#include <algorithm>
#include <iostream>
using namespace std;
typedef struct {
int u;
int v;
int w;
} Edge;
int f[201];
bool cmp(Edge e1, Edge e2) {
return e1.w < e2.w; }
void init() {
for (int i = 0; i <= 200; i++) {
f[i] = i;
}
}
int find(int x) {
if (x == f[x])
return x;
return f[x] = find(f[x]);
}
int merge(int x, int y) {
int a = find(x);
int b = find(y);
if (a != b) {
f[b] = a;
return 1;
}
return 0;
}
int main() {
int n, e;
while (cin >> n >> e) {
Edge edge[e];
init();
for (int i = 0; i < e; i++) {
cin >> edge[i].u >> edge[i].v >> edge[i].w;
}
sort(edge, edge + e, cmp);
int cnt = 0, sum = 0;
for (int i = 0; i < e; i++) {
if (merge(edge[i].u, edge[i].v)) {
cnt++;
sum += edge[i].w;
}
if (cnt == n - 1)
break;
}
if (cnt == n - 1)
cout << sum << endl;
else
cout << "There is no minimum spanning tree." << endl;
}
return 0;
}