数组模拟实现并查集 理论篇

第二章 数据结构（二） - AcWing

集合我们可以用树的形式来表示，假设现在有两个集合A,B：

比如我们想查找2,3这两个节点是否是同一个集合就需要判断2,3这两个节点是否是同一个root节点。那么就需要判断2,3节点是否是同一个根节点，我们判断的方法是让2,3节点去找它们的父节点，找到之后判断父节点是否是root节点，如果不是那就继续往上走。这个过程中会有一个优化，即路径压缩，下次再找root节点的时候就不需要一层层往上找了，直接可以找到跳到root节点，图示如下：

836. 合并集合 - AcWing题库

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int p[N];

//判断是否在同一个集合
int find(int x)
{
    if(p[x]!=x)p[x]=find(p[x]);  //如果x不是根 那么就让它向上查找，找它的根
    return p[x]; //返回x的根节点
}
int main()
{
    int n,m;cin>>n>>m;
    
    for(int i=1;i<=n;i++)p[i]=i;  

  while(m--)
    {   
        char op;
        int a,b;
        scanf(" %c%d%d",&op,&a,&b);
      
 
        if(op=='M')

            p[find(a)]=find(b);  //让b集合的根成为a集合根的根节点

       else  
        {
            //cout<<"op: find(a)： find(b) "<<op<<" "<<find(a)<<" "<<find(b)<<endl;
            if(find(a)==find(b)) //是同一个根说明是同一个集合
            cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }
    }
    return 0;
}