算法练习之并查集(路径压缩)

[无你想你]

#include<iostream>
using namespace std;
const int N=100010;
int n,m;
int p[N],size[N];
 
int find(int x)//这里实现了路径压缩
{
    if(p[x]!=x)p[x]=find(p[x]);
    return p[x]; 
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;++i)
    {
        p[i]=i;
        size[i]=1;
    }
    while(m--)
    {
        char op[5];
        int a,b;
        scanf("%s",op);
        if(op[0]=='C')
        {
            scanf("%s",op);
            if(find(b)==find(a))continue;
            size[find(b)]+=size[find(a)];
            p[find(a)]=find(b);
        }
        else if(op[1]=='1')
        {
            scanf("%d%d",&a,&b);
            if(find(a)==find(b))puts("Yes");
            else puts("No");
        }
        else
        {
            scanf("%d",&a);
            printf("%d\n",size[find(a)]);
        }
    }
    return 0;
}

[无你想你]

#include
using namespace std;
const int N=100010;
int p[N],cnt[N];

int find(int x)
{
// 寻找父节点
if(p[x]!=x)p[x]=find(p[x]);
return p[x];
}
void Union(int a,int b)
{
int pa=find(a),pb=find(b);
if(pa!=pb)
{
cnt[pb]+=cnt[pa];//把pa中连通块的个数加入到pb中
p[pa]=pb;
}

}

int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;++i)p[i]=i,cnt[i]=1;

while(m--)
{
    string op;
    int a,b;
    cin>>op;
    
    if(op=="C")
        {
            cin>>a>>b;
            Union(a,b);
        }
    else if(op=="Q1")
        {
            cin>>a>>b;
            if(find(a)==find(b))puts("Yes");
            else puts("No");
        }
    else
    {
     cin>>a;
     cout<<cnt[find(a)]<<endl;
    }
}
return 0;

}