并查集-连通块中点的数量，记录每个集合的大小

考察点：记录每个集合的大小，这个属性可以绑定到根节点上。
原题链接：https://www.acwing.com/problem/content/description/839/

#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--)
    {
        int a,b;
        char op[2];
        scanf("%s",op);
        
        if(op[0]=='C')
        {
            scanf("%d%d",&a,&b);
            if(find(a)==find(b)) continue;
            // 把a合并到b的集合中来。数量加到b的祖宗的头上
            s[find(b)]+=s[find(a)];
            // 把a合并到b，需要先加数量才能合并。
            p[find(a)]=find(b);
        }
        else if(op[1]=='1')
        {
            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;
}