【并查集】连通块中点的数量

题目

解释

• 点与点之间的连通相当于合并集合，所以这里使用并查集来做
• size数组用来存放当前始祖节点为根的树节点个数

代码段

#include<iostream>
#include<string>
using namespace std;
const int N=1e5+10;
int p[N],cnt[N];
int n,m;
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;
        cnt[i]=1;
    }
    while(m--)
    {
         char op[2];
         int a,b;
         cin>>op;
        if(op[0]=='C')
        {
        cin>>a>>b;
        if(find(a)==find(b))continue;
        //如果已经连通，则继续
        cnt[find(b)]+=cnt[find(a)];
        p[find(a)]=find(b);
        //如果未连通，则b成为a的祖先，将a和它的子节点数量增加到b中
        }
        else if(op[1]=='1')
        {
            cin>>a>>b;
            if(find(a)==find(b))cout<<"Yes"<<endl;
            //如果祖先相同，则连通
            else
            cout<<"No"<<endl;
        }
        else
        {
            cin>>a;
            cout<<cnt[find(a)]<<endl;
            //返回连通图的点个数
        }
    }
}