两个人到底连不连得通

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

补全 connected：判断两个人在不在同一块里。把两个判断拼起来输出：阿岚和南风、阿岚和北辰，中间用 / 隔开。

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

提交你的答案

请登录后提交答案。

去登录

← 用并查集数出有几块一个够用的图至少要有什么 →

更多题目

代码编辑器

语言： 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;
}

bool connected(vector<int>& p, int a, int b) {
    // TODO: 两个人的根一样就算连通
    return false;
}

int main() {
    auto p = init_uf();
    for (auto [a, b] : EDGES) unite(p, a, b);
    cout << boolalpha << connected(p, 0, 3) << "/" << connected(p, 0, 4) << endl;
}

本次输入：

输出：

👩‍🏫

💬 题目评论

两个人到底连不连得通

更多题目

全部评论