CCF CSP 竞赛试题——数据中心（201812-4）

最新推荐文章于 2022-09-25 22:03:43 发布

原创最新推荐文章于 2022-09-25 22:03:43 发布 · 306 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

CCF CSP 竞赛试题专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一种使用并查集实现最小生成树的算法。通过实例演示了如何利用权值最小的边连接节点，形成包含所有节点的树。代码示例清晰展示了并查集的查找与合并操作。

并查集
和root是哪个节点毫无关系
题意为【用权值尽可能小的边，将这n个节点连起来】

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

vector<int> nodes;

int fd(int x) { return nodes[x] = nodes[x] == x ? x : fd(nodes[x]); }

int main() {
    int n, m, root;
    cin >> n >> m >> root;

    for (int i = 0; i <= n; ++i) nodes.push_back(i);

    vector<int> e(3);
    vector<vector<int> > edge;
    while (m--) {
        cin >> e[2] >> e[1] >> e[0];
        edge.push_back(e);
    }
    sort(edge.begin(), edge.end());
    for (int i = 0; i < edge.size(); ++i) {
        int x = fd(edge[i][1]), y = fd(edge[i][2]);
        if (x != y) {
            if (--n == 1) {
                cout << edge[i][0] << endl;
                break;
            }
            nodes[x] = y;
        }
    }
    return 0;
}