在原来并查集的基础上增加一个Size数组,Size的初始化是必须先每个元素初始化为1
Size只对根节点有效,比如Size[find(1)]就是找1的祖先节点,然后访问祖先节点的个数。当我们联通a点和b点时,如果已经是联通状态了,那么无需继续执行,跳过。否则,那么我们可以把a点加入到b点,然后只对b点所在祖先节点的数量加上a祖先节点的数量,也就是Size[find(b)]+=Size[find(a)];
#include<iostream>
#define LEN 100086
#include<string>
using namespace std;
int n,m,p[LEN],Size[LEN];
string op;
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;
Size[i]=1;
}
while(m--){
cin>>op;
int a,b;
if(op=="C"){
cin>>a>>b;
if(find(a)==find(b))
continue;
Size[find(b)]+=Size[find(a)];
p[find(a)]=find(b);
}else if(op=="Q1"){
cin>>a>>b;
bool isIn=(find(a)==find(b));
cout<<(isIn?"Yes":"No")<<endl;
}else if(op=="Q2"){
cin>>a;
cout<<Size[find(a)]<<endl;
}
}
return 0;
}