837. 连通块中点的数量 AcWing

原题链接

并查集模板题

当两个点互相可达,我们称它们连通.本题判断连通点的个数,就是判断同一集合下点的个数,一棵树下子节点各不同,因此用数组sizes记录下标为根节点的点的个数

#include <iostream>
using namespace std;
const int N = 1e5+10;
int p[N],sizes[N];
int find(int x)
{
    if(p[x]!=x) p[x] = find(p[x]);
    return p[x];
}
int main()
{
    int n,m;
    char op[3];
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) { p[i] = i;sizes[i] =1; }
    while(m--){
        scanf("%s",op);
        if(op[0]=='C') {
            int x,y;
            scanf("%d%d",&x,&y);
            if(find(x)==find(y)) continue;
            sizes[find(y)]+=sizes[find(x)];
            p[find(x)] = find(y);
        }
        else if(op[0]=='Q'){
            if(op[1]=='1'){
                int x,y;
                scanf("%d%d",&x,&y);
                if(find(p[x])==find(p[y])) printf("Yes\n");
                else printf("No\n");
            }else if(op[1]=='2'){
                int x; scanf("%d",&x);
                printf("%d\n",sizes[find(x)]);
            }
        }
    }
    return 0;
}