并查集

837. 连通块中点的数量 - AcWing题库

AC代码：
 

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 100010;
int n,m;
int p[N],c[N];

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;
        c[i]=1;
    }
    
    while(m--)
    {
        string op;
        int a,b;
        cin>>op;
        
        if(op=="C")
        {
             cin>>a>>b;
             if(find(a)!=find(b))
             {
                 //注意这里的顺序不要写反
                 c[find(b)]+=c[find(a)];
                 p[find(a)]=find(b);
             }
           
        }
        else if(op=="Q1")
        {
            cin>>a>>b;
            if(find(a)==find(b))cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }
        else
        {
            cin>>a;
            cout<<c[find(a)]<<endl;
        }
        
    }
    
    return 0;
}

相关解释：

这两个语句顺序不要写反，如果写反的话：

这样写是错误的。因为我们是先把a的祖先改为b的祖先，那么此时计算数量的话，c[b的祖先]+=c[a的祖先]//因为有上一句，所以这里相当于c[b的祖先]+=c[b的祖先],那么显然是错的。

所以这里计算数量的时候，要先加，然后在更新。