并查集的作用、原理、实现及应用

并查集

并查集的作用

1. 将两个集合合并
2. 询问两个元素是否在一个集合当中

基本原理

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

2. 如何判断树根：if (p[x] == x) 除了根节点之外p[x]都不等于x

3. 如何求x的集合编号：while (p[x] != x) x = p[x] 优化：路径压缩

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

836. 合并集合 - AcWing题库

#include <iostream>

using namespace std;

const int N = 100010;
int n, m;
int p[N];

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

int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i++){
        p[i] = i;
    }    
    while(m--){
        char op[2];
        int a, b;
        cin >> op >> a >> b;
        if(op[0] == 'M'){
            p[find(a)] = find(b);
        }else{
            if(find(a) == find(b)){
                cout << "Yes" << endl;
            }else{
                cout << "No" << endl;
            }
        }
    }
    return 0;
}

837. 连通块中点的数量 - AcWing题库

增加维护集合的数量

#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()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i++){
        p[i] = i;
        Size[i] = 1;
    }
    while(m--){
        char op[5];
        int a, b;
        cin >> op;
        if(op[0] == 'C'){
            cin >> a >> b;
            if(find(a) == find(b)){
                continue;
            }
            Size[find(b)] += Size[find(a)];
            p[find(a)] = find(b);
        }else if(op[1] == '1'){
            cin >> a >> b;
            if(find(a) == find(b))cout << "Yes" << endl;
            else cout << "No" << endl;
        }else{
            cin >> a;
            cout << Size[find(a)] << endl;
        }
    }
    
    return 0;
}