用并查集数出有几块

👁️ 0 人浏览 💬 0 人评论 ❤️ 添加收藏

合并完所有关系之后，补全代码数出一共有几块。做法：每个人各自 find 一次，数出有几个不同的根。

（本题用 g++ -std=c++17 -O0 编译。）

提交你的答案

请登录后提交答案。

去登录

← 写一个 unite 两个人到底连不连得通 →

更多题目

代码编辑器

语言： Ctrl + Enter 运行

#include <iostream>
#include <string>
#include <utility>
#include <vector>
using namespace std;

// 五个人：0 阿岚、1 小满、2 阿泰、3 南风、4 北辰
const vector<string> NAMES = {"阿岚", "小满", "阿泰", "南风", "北辰"};
// 四段关系（无向）：阿岚-小满、阿岚-阿泰、小满-阿泰、阿泰-南风
const vector<pair<int, int>> EDGES = {{0, 1}, {0, 2}, {1, 2}, {2, 3}};

int id_of(const string& name) {
    for (int i = 0; i < (int)NAMES.size(); i++) {
        if (NAMES[i] == name) return i;
    }
    return -1;
}

vector<int> init_uf() {
    vector<int> p(NAMES.size());
    for (int i = 0; i < (int)p.size(); i++) p[i] = i;   // 每人先自成一块
    return p;
}

int find_root(vector<int>& p, int x) {
    while (p[x] != x) {
        p[x] = p[p[x]];      // 路径压缩：顺手往上挂一层
        x = p[x];
    }
    return x;
}

void unite(vector<int>& p, int a, int b) {
    int ra = find_root(p, a), rb = find_root(p, b);
    if (ra != rb) p[ra] = rb;
}

int main() {
    auto p = init_uf();
    for (auto [a, b] : EDGES) unite(p, a, b);
    int k = 0;
    // TODO: 数一数有几个人的根就是他自己（每一块恰好有一个根）
    cout << k << endl;
}

本次输入：

输出：

👩‍🏫

💬 题目评论

用并查集数出有几块

更多题目

全部评论