ccf 201412-4 最优灌溉 100

最小生成树Kruskal算法

最新推荐文章于 2020-09-05 11:59:54 发布

Bendaai

最新推荐文章于 2020-09-05 11:59:54 发布

阅读量385

点赞数

CC 4.0 BY-SA版权

文章标签： ccf

本文链接：https://blog.youkuaiyun.com/Bendaai/article/details/77790532

本文介绍了一种实现最小生成树(MST)的Kruskal算法，通过使用C++编程语言并结合输入输出流、算法库及向量容器等标准模板库组件，实现了边的定义、比较操作符、寻找根节点、合并节点等功能。该算法首先读取顶点数量和边的数量，然后接收每条边的两个顶点及其权重，并将这些边存储在一个向量中，接着对所有边按权重进行排序，最后遍历排序后的边集合，对于每一条边，如果它连接的两个顶点不处于同一个连通分量，则将这条边加入到MST中并合并这两个顶点所在的集合。

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int pre[1005];
class edge {
public:
    int a, b, cost;
    bool operator<(const edge b)const {
        return cost < b.cost;
    }
};
bool compare(const edge a, const edge b) {
    return a.cost < b.cost;
}
vector<edge> all;
int findroot(int i) {
    if (pre[i] != i) pre[i] = findroot(pre[i]);
    return pre[i];
}
void addv(int i, int j) {
    int x = findroot(i);
    int y = findroot(j);
    pre[x] = pre[y] = min(x, y);
}
int main() {
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; ++i)
        pre[i] = i;
    edge a;
    while (m--) {
        cin >> a.a >> a.b >> a.cost;
        all.push_back(a);
    }
    sort(all.begin(), all.end(),compare);
    int re = 0, cnt = 1;
    for (int i = 0; i < all.size(); ++i) {
        if (findroot(all[i].a) != findroot(all[i].b)) {
            addv(all[i].a, all[i].b);
            re += all[i].cost;
            if (++cnt == n) break;
        }

    }
    cout << re << endl;
}