并查集的理解与代码实现

并查集的作用：

        1、将两个集合合并

        2、判断两个元素是否在一个集合当中

基本原理： 每个集合用一棵树来表示，树根的编号就是整个集合的编号，每个节点存储它的父节点，p[x]表示x的父节点。

问题1： 如何判断是树根： if(p[x]==x)

问题2： 如何求x的集合编号： while(p[x] != x) x = p[x];

问题3： 如何合并两个集合 ： px是x的集合编号，py是y的集合编号。p[x]=y

存在的一个复杂度问题： 在求x编号的时候，每一次都要从底部进行遍历到根节点，每次遍历的复杂度都是树的深度。所以采用路径压缩的方式，在第一次遍历之后呢，把除了根节点之外的所有节点都直接连接到根节点上上面，下一次遍历找x编号的时候复杂度就是O(1)

优化： 采用路径压缩

#include <bits/stdc++.h>
using namespace std;

const int N = 100010;
int n,m;
int p[N]; //用来存放每个节点的父节点

int find(int x){
    if(p[x]!=x) p[x] = find(p[x]);  //如果x不是根节点，递归，直到x是根节点
    return p[x];  //返回祖宗节点
}


int main(){
    scanf("%d%d",&n,&m);
    for(int i =1;i<=n;i++)
        p[i]=i;//初始化的时候先将每个节点的父节点赋值为自己

    while(m--){
        char op[2];
        int a,b;
        scanf("%s%d%d",op,&a,&b);
        //合并两个集合
        if(op[0] == 'M') p[find(a)] = find(b);   //将b集合的祖宗节点，也就是find(b) 复制给集合a的祖宗节点的父亲
        else{  //查看两个集合是否在一个大集合内
            if(find(a) == find(b)) puts("yes");   //如果两个根节点的父节点相等
            else puts("no");
        }

    }
    return 0;
}