没有rank优化的简单并查集| #第一题#

第一题

https://www.nowcoder.com/practice/7c29cdfa28274c86afc9e88c07448a10

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

const int N = 1e6 + 10;
int f[N];//并查集
set<int> nodes;//有哪些节点

//查找n的根
int find(int n) {
    if (f[n] == n)
        return  n;
    return find(f[n]);
}

//简单合并,不涉及高度
void merge(int a, int b) {
    int r1 = find(a), r2 = find(b);
    if (r1 != r2)
        f[r2] = r1;
}

void init() {
    for (int i = 0; i < N; i++)
        f[i] = i;
}

int count() {
    set<int> s;
    for (int node:nodes) {
        int root = find(node);
        s.insert(root);
    }
    return s.size();
}
int main() {
    int x, y;
    init();
    while (cin >> x >> y) {
        nodes.insert(x);
        nodes.insert(y);
        merge(x, y);
    }
    cout << count() << endl;
}
// 64 位输出请用 printf("%lld")

全部评论

相关推荐

11-04 14:10
东南大学 Java
_可乐多加冰_:去市公司包卖卡的
点赞 评论 收藏
分享
11-05 07:29
贵州大学 Java
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务