837. 连通块中点的数量并查集

最新推荐文章于 2024-09-08 15:12:17 发布

我也念过晚霞

最新推荐文章于 2024-09-08 15:12:17 发布

阅读量223

点赞数 2

分类专栏： AcWing 文章标签：算法数据结构图论

本文链接：https://blog.csdn.net/2303_79132118/article/details/136871437

版权

AcWing 专栏收录该内容

254 篇文章 0 订阅

订阅专栏

这篇文章介绍了使用C++实现的一个树形数据结构，涉及节点合并、查找根节点以及查询操作。通过`find`函数维护树的连通性，并在合并节点时更新父节点的大小。

摘要由CSDN通过智能技术生成

要先计算再合并

1.先合并，那么a的父节点变成了b的父节点
2.计算是b的父节点记录的size值加上a的父节点记录的size值，这时候不就变成了两个b父节点的size和吗？

#include <iostream>

using namespace std;

const int N = 100010;

int n, m;
int p[N], s[N];

int find(int x){
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main(){
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++){
        p[i] = i;
        s[i] = 1;
    }

    while(m --)
    {
        char op[5];
        int a, b;
        scanf("%s", op);
        if(op[0] == 'C'){ //合并
            scanf("%d%d", &a, &b);
            if(find(a) == find(b)) continue;
            s[find(b)] += s[find(a)];
            p[find(a)] = find(b);
        }
        else if(op[1] == '1'){ //Q1
            scanf("%d%d", &a, &b);
            if(find(a) == find(b)) puts("Yes");
            else puts("No");
        }
        else{
            scanf("%d", &a);
            printf("%d\n", s[find(a)]);
        }
    }

    return 0;
}

。