AcWing 837.连通块中点的数量(并查集）

题目

---

思路

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;    
}