思路
int siz[maxn];
- 和上一个题几乎一样,就是多了个Q2操作,求这个点所在的连通块中的点的个数,所以只需要在上一次思路的基础上,另外开辟一个数组
siz[find(b)]+=siz[find(a)];
这是更新连接后的集合的元素的个数,
p[find(a)]=find(b);
把b节点所在的树连接到a节点所在的树的根节点上
代码
#include<iostream>
#include<algorithm>
using namespace std;
int const maxn=100010;
int p[maxn],siz[maxn];/siz[]存的是每个集合有多少元素,当求Q2时会用到
int find(int x)
{
if(p[x]!=x) //如果不是根节点,那么就找到根节点
p[x]=find(p[x]);
return p[x];//返回根节点
}
int main(void)
{
int n,m;
cin>>n>>m;
for(int i = 1; i <= n; i ++) p[i]=i,siz[i]=1;//一开始每个点都是孤立的,所在一个集合里
while(m--)
{
string op;
cin>>op;
int a,b;
if(op[0]=='C')
{
cin>>a>>b;
if(find(a)!=find(b)) //如果两个点的根节点不一样
{
siz[find(b)]+=siz[find(a)]; ///这是更新连接后的集合的元素的个数,
p[find(a)]=find(b);// 把b节点所在的树连接到a节点所在的树的根节点上
}
}
else if(op=="Q1")
{
cin>>a>>b;
if(find(a)==find(b)) //如果根节点相同,也就是在一个集合里
cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
else if(op=="Q2")
{
cin>>a;
cout<<siz[find(a)]<<endl;
}
}
return 0;
}