7-5 最小支撑树 (10分)(Kruskal算法)

给定一个包含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;
}

猜你喜欢

转载自blog.csdn.net/weixin_45845039/article/details/113339432