AcWing-AcWing 837. 连通块中点的数量

在原来并查集的基础上增加一个Size数组，Size的初始化是必须先每个元素初始化为1
Size只对根节点有效，比如Size[find(1)]就是找1的祖先节点，然后访问祖先节点的个数。当我们联通a点和b点时，如果已经是联通状态了，那么无需继续执行，跳过。否则，那么我们可以把a点加入到b点，然后只对b点所在祖先节点的数量加上a祖先节点的数量，也就是Size[find(b)]+=Size[find(a)];

#include<iostream>
#define LEN 100086
#include<string>
using namespace std;
int n,m,p[LEN],Size[LEN];
string op;
int find(int x){
    if(p[x]!=x)
        p[x]=find(p[x]);
    return p[x];
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;++i){
         p[i]=i;
         Size[i]=1;
    }
    while(m--){
        cin>>op;
        int a,b;
        if(op=="C"){
            cin>>a>>b;
            if(find(a)==find(b))
                continue;
            Size[find(b)]+=Size[find(a)];
            p[find(a)]=find(b);
        }else if(op=="Q1"){
            cin>>a>>b;
            bool isIn=(find(a)==find(b));
            cout<<(isIn?"Yes":"No")<<endl;
        }else if(op=="Q2"){
            cin>>a;
            cout<<Size[find(a)]<<endl;
        }
    }
    return 0;
}