16 并查集

/*
学acwing的算法基础课学来的，喜欢的话多多支持呀。
*/
//最裸的并查集
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int n,m,p[N];
int size[N];//开辟一个size来维护一个新的集合的个数
int find(int x)//返回x的祖宗结点+路径压缩
{
    if(p[x]!=x) p[x]=find(p[x]);//递归来寻找
    return p[x];//返回祖宗结点
}
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m;
  for(int i=1;i<=n;i++) {
        p[i]=i;//初始化每个祖宗结点
        size[i]=1;//初始化每个刚开始的祖宗结点为1
  }
    while(m--)
   {
    char op[2];
    int a,b;
    cin>>op;//读入一个操作
    if(op[0]=='C')
    {
        cin>>a>>b;
        if(find(a)==find(b))continue;//假如是一个集合了，则不用合并
        size[find(b)]+=size[find(a)];//a合并到b的话，就把a的个数加到b中
        p[find(a)]=find(b);//假如是合并，则把其中一个的祖宗连接到另一个祖宗中
    }
    else if(op[1]=='1')//假如是匹配
    {
       cin>>a>>b;
      if(find(a)==find(b)) puts("YES");//匹配成功
      else puts("NO");  //失败
    }
    else{
        cin>>a;
        cout<<size[find(a)]<<endl;//输出该点祖宗的仔个数
    }
   }
    return 0;
}
/*到达胜利之前无法回头*/