【刷题】数据结构——并查集【模板】

并查集用于快速判断两个元素是否在同一个集合中。

用树的形式来表达各个集合的关系，一棵树上的所有点代表是同一个集合。
P[i]表示编号为i节点的祖先，初始值都是自己。
合并两节点的集合时，只需在两个节点的祖先建立一条边，即将一个结点祖先指向另一个节点祖先，也就是P[find(a)]=find(b)。
判断两结点是否在同个集合，只需看两个节点是否是同一个祖先。

find(x)是找x的祖先，通过递归find(p[x])进行查找，直到找到最顶层的祖先(x==p[x])时停止。

同时利用递归可以进行路径压缩，将P[x]直接指向最顶层的祖先。

#include <iostream>
using namespace std;
const int N = 100005;
int n, m, p[N];

void init() {
    for (int i = 1; i <= n; i ++ ) p[i] = i;
}

int find(int x) { // 返回x的祖先 
    if (p[x] != x) p[x] = find(p[x]); // 路径压缩
    return p[x];
}


void merge(int a, int b) {
    p[find(a)] = find(b);
    // 此处不能写p[a] = find(b)，否则只是将a合并到b所在集合B上，而不是将a所在的集合合并到B上
}

int main() {
    scanf("%d%d", &n, &m);
    init();
    for (int i = 1; i <= m; i ++ ) {
        char op[2];
        int a, b;
        scanf("%s%d%d", op, &a, &b);
        if (op[0] == 'M') {
            merge(a, b);
        }
        else {
            if (find(a) == find(b)) printf("Yes\n");
            else printf("No\n");
        }
    }
    return 0;
}