并查集(集合合并,路径压缩优化)

并查集(路径压缩优化)

摘自acwing算法模板

并查集

并查集的作用：
1.两个集合合并
2.询问两个集合是否在同一个集合中

怎么实现路径压缩？
如果x不是祖宗结点，就让父亲结点 = 祖宗结点 ， 最后返回父亲结点

怎么实现集合合并
让a祖宗的结点的父亲等于b结点的结点

代码

#include<iostream>

using namespace std ;

const int N = 100010 ;

int p[N];

int find(int x) // 集合合并+路径压缩
{
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main()
{
    int n,m;
    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 == 'M') p[find(a)] = find(b);
        else
        {
            if(find(a) == find(b)) puts("Yes");
            else puts("No");
        }
    }
    
    return 0;
}