837. 连通块中点的数量

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;

int n,m;
int f[N],cnt[N];//cnt[i]表示该集合i的节点数量

//查询x所属集合
int find(int x){
    if(f[x]!=x) f[x]=find(f[x]);
    return f[x];
}

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        f[i]=i;
        cnt[i]=1;
    }
    
    //执行操作
    string op;
    int a,b;
    while(m--){
        cin>>op>>a;
        if(op=="C"){
            cin>>b;
            if(find(a)==find(b)) continue;//在同一个集合里
            cnt[find(b)]+=cnt[find(a)];//结点数相加，只维护根结点cnt
            f[find(a)]=find(b);
        }
        else if(op=="Q1"){
            cin>>b;
            if(find(a)==find(b)) cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }
        else{
            cout<<cnt[find(a)]<<endl;
        }
    }
    return 0;
}