俩行代码也太优雅了吧,递归~
#include<iostream>
#include<cstring>
using namespace std;
const int N=10010;
int n,m;
int p[N];//存储数的父节点
int asize[N];//存储集合大小
int find(int x)
{
if(p[x]!=x)p[x]=find(p[x]);//将父节点更新为祖宗节点
return p[x];
}
//合并,将a接入b
void add(int a,int b)
{
if(find(a)==find(b))return;
else(asize[find(b)]+=asize[find(a)]);
p[find(a)]=find(b);return;
}
//询问是否在同一集合
int query(int a,int b)
{
if(find(a)==find(b))return 1;
return 0;
}
//返回该数所在集合的元数总数
void sum(int a)
{
cout<<asize[find(a)]<<endl;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
p[i]=i;
asize[i]=1;
}
while(m--)
{
char op[5];
int a,b;
cin>>op;
if(op[0]=='C')
{
cin>>a>>b;
add(a,b);
}
else if(op[0]=='a')
{
cin>>a>>b;
if(query(a,b)==1)puts("yes");
else puts("no");
}
else
{
cin>>a;
sum(a);
}
}
return 0;
}