（并查集+树的大小）acwing 837. 连通块中点的数量

最新推荐文章于 2024-10-31 19:07:52 发布

岁忧

最新推荐文章于 2024-10-31 19:07:52 发布

阅读量122

点赞数 1

分类专栏： AcWing 文章标签： c语言算法并查集

本文链接：https://blog.csdn.net/weixin_46028214/article/details/116561943

版权

AcWing 专栏收录该内容

328 篇文章 3 订阅

订阅专栏

837. 连通块中点的数量

题目链接https://www.acwing.com/problem/content/description/839/
题目：
在这里插入图片描述

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;
const int N=1e5+10;
int p[N],cnt[N];
int findd(int u){
    if(p[u]!=u) p[u]=findd(p[u]);
    return p[u];
}
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=1;i<N;i++){
        p[i]=i;
        cnt[i]=1;
    }
    char op[3];
    int a,b,x,y;
    while(m--){
        scanf("%s",op);
        //cout<<op<<endl;
        if(!strcmp(op,"C")){
            scanf("%d%d",&a,&b);
            x=findd(a),y=findd(b);
            if(x!=y){//这里是判断a,b是否在同一颗树下
                p[x]=y;
                cnt[y]+=cnt[x];
            }
        }else if(!strcmp(op,"Q1")){
            scanf("%d%d",&a,&b);
            x=findd(a),y=findd(b);
            if(x==y) printf("Yes\n");
            else printf("No\n");
        }else{
            scanf("%d",&a);
            x=findd(a);
            printf("%d\n",cnt[x]);
        }
    }
    return 0;
}