AcWing 837. 连通块中点的数量
原题链接
简单
#include<iostream>
using namespace std;
const int N=1e5+10;
int p[N],cnt[N];
int n,m;
int find(int x)
{
if(p[x]!=x)
return p[x]=find(p[x]);
return p[x];
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
p[i]=i;
cnt[i]=1;
}
while(m--)
{
int a,b;
char op[3];
scanf("%s",op);
if(op[0]=='C')
{
scanf("%d%d",&a,&b);
if(find(a)!=find(b))
{
cnt[find(b)]+=cnt[find(a)];
p[find(a)]=find(b);
}
}
else
{
if(op[1]=='1')
{
scanf("%d%d",&a,&b);
if(find(a)==find(b))
printf("Yes\n");
else
printf("No\n");
}
else
{
scanf("%d",&a);
printf("%d\n",cnt[find(a)]);
}
}
}
return 0;
}