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

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

原题链接

AcWing 837. 连通块中点的数量

算法标签

并查集

思路

见代码

代码

#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 100005;
// p存储节点祖宗节点 s存储所属连通块节点数
int p[N], s[N];
inline int read(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
// 查询祖宗节点
int find(int x){
    if(p[x]!=x){
        return p[x]=find(p[x]);
    }else{
        return p[x];
    }
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n=read(), m=read();
    rep(i, 1, n+1){
        p[i]=i;
        // 节点数初始化为1
        s[i]=1;
    }
    while(m--){
        char op[5];
        scanf("%s", op);
        if(op[0]=='C'){
            int a=read(), b=read();
            if(find(a)!=find(b)){
                s[find(b)]+=s[find(a)];
                p[find(a)]=find(b);
            }
        }else if(op[0]=='Q'&&op[1]=='1'){
            int a=read(), b=read();
            if(find(a)==find(b)){
                puts("Yes");
            }else{
                puts("No");
            }
        }else{
            int a=read();
            printf("%lld\n", s[find(a)]);
        }
    }
    return 0;
}

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈