算法基础课——第二章 数据结构（一）

并查集

概念

并查集是一种可以动态维护若干个不重叠的结合,并且支持合并与查询的数据结构，也就是擅长维护各种各样的具有传递性质的关系.

基本操作

并查集主要的操作分为两类

1. find（查找）操作：查询该元素属于哪个集合。
2. merge（合并）操作：将两个集合合并为一个集合。

算法步骤

初始化操作：使用树形结构存储每个集合,刚开始的时候每个点就是一个独立的集合,且这个集合树的树根就是本身.

//代码实现
for(int i = 1;i<= n;i ++ ) p[i] = i;
//p数组代表其父节点，p即为parent

优化方法（路径压缩）

如不优化路径，则当树状数组深度非常大时，查找元素的祖宗节点将变得非常慢。优化后的时间复杂度则直接达到O(1)，效率大大提高。

//代码实现
int find(int x){
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];//直到找到祖宗结点
}

下图为路径压缩前后直观对比图。

优化前：

优化后：

例题展示

例题一：

原题链接

解题思路

• 将每次输入的数据组成（加入）一个无向图，输入完之后判断节点1所在的该图的节点个数。如果求得的结点个数与输入的结点个数相同则为连通的无向图，反之则为不连通。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
#include <unordered_map>
#define fi first
#define se second
#define VI vector<int>
#define IOS ios::sync_with_stdio(false),cin.tie(nullptr)
#define endl '\n'
#define PII pair<int ,int>
#define PIS pair<int ,string>
#define PSI pair<string ,int>
#define PSS pair<string ,string>
#define int long long
#define mod 1000000010

using namespace std;

VI p(1010),cnt(5010);

int find(int x){
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

signed main(){
    IOS;
    int n,m;
    while(cin >> n >> m){
        for(int i = 1;i<= n;i ++ ) cnt[i] = 1,p[i] = i;

        while(m--){
            int a,b;cin >> a >> b;
            int x = find(a),y = find(b);
            if(x != y){
                p[x] = y;
                cnt[y] += cnt[x];
            }            
        }

        if(cnt[find(1)] == n) cout << "YES" << endl;
        else cout << "NO" << endl;
    }

    return 0;
}

例题二：

原题链接

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
#include <unordered_map>
#define fi first
#define se second
#define VI vector<int>
#define IOS ios::sync_with_stdio(false),cin.tie(nullptr)
#define endl '\n'
#define PII pair<int ,int>
#define PIS pair<int ,string>
#define PSI pair<string ,int>
#define PSS pair<string ,string>
#define int long long
#define mod 1000000010

using namespace std;
int p[100010];
int n,m;

int find(int x){
    if(x != p[x]) p[x] = find(p[x]);
    return p[x];
}

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