连通块中点的数量 --并查集（c++）

5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5

Yes
2
3

#include <bits/stdc++.h>

using namespace std;

const int N = 100010;
int n,m;
int p[N],siz[N];
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;;
        siz[i] = 1;
    }
    while(m--)
    {
        char op[2];
        int a,b;
        scanf("%s",op);  
    
        if(op[0] == 'C')  
        {
            scanf("%d%d",&a,&b);
            if(find(a) == find(b)) continue;
            siz[find(b)] += siz[find(a)];
            p[find(a)] = find(b);  //使a的祖宗节点的父节点等于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",siz[find(a)]);
        }
    }
    return 0;
}