AcWing 837. 连通块中点的数量

#include<algorithm>
#include<iostream>
using namespace std;
const int N=1e5+10;
int p[N];
int n,m;
int sizee[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;
        sizee[i]=1;
    }
    while(m--)
    {
        string s;
        int a,b;
        cin>>s;
        if(s=="C")
        {
            cin>>a>>b;
            if(find(a)==find(b))//如果为同一集合，则继续
            continue;
            sizee[find(b)]+=sizee[find(a)];//不是同一集合时，把a所在集合
            						       //的数量加	入到b所在集合中
            p[find(a)]=find(b);//把a所在集合加到b所在集合
            
        }
        else if(s=="Q1")
        {
            cin>>a>>b;
            if(find(a)==find(b))
            cout<<"Yes"<<endl;
            else
            cout<<"No"<<endl;
        }
        else
        {
            cin>>a;
            cout<<sizee[find(a)]<<endl;//a所在集合的点的数量
        }
    }
    return 0;
}