[数据结构]并查集(系统整理版)

基础用法

cpp 复制代码
int p[N];

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

int main(){
//初始化
    for(int i=1;i<=n;i++) p[i]=i;
}

合并

cpp 复制代码
void merge(int a,int b){
    int aa=find(a),bb=find(bb);
    if(aa!=bb){
        p[aa]=bb;
    }
}

查询是否联通

cpp 复制代码
bool is_connected(int a,int b){
    int aa=find(a),bb=find(bb);
    if(aa==bb) return 1;
    else return 0;
}

如何维护联通块数量

初始化cnt为元素个数n

每次合并时 cnt--

最后cnt即为最后的连通块个数

cpp 复制代码
int cnt=n;
void merge(int a,int b){
    int aa=find(a),bb=find(bb);
    if(aa!=bb){
        p[aa]=bb;
    }
    cnt--;
}

如何维护每个连通块的元素个数

维护一个size数组s 初始化为1
for(int i=1;i<=n;i++) s[i]=1;

  • size数组不能定义为int size[N]size会与cpp内部的关键字混淆
cpp 复制代码
int s[N];
int cnt=n;

void merge(int u,int v){
    int uu=find(u),vv=find(v);
    if(uu!=vv){
        if(s[uu]>s[vv]){
            p[vv]=uu;
            s[uu]+=s[vv];
        }else{
            p[uu]=vv;
            s[vv]+=s[uu];
        }
        cnt--;
    }
}

按秩合并

每次合并把元素少的连通块合并到元素多的去

因为并查集类似树形结构

这样合并能时树的高度增长相对少

减少路径压缩次数 提高查询效率

如何像STL一样使用并查集

借鉴了LeetCode的并查集模板

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;


class UnionFind{
private:
    vector<int>p;
    vector<int>s;
    int cnt=0;
public:
    UnionFind(int n): p(n+1),s(n+1,1){//多开一个空间 同时适用于0-based&1-based
        iota(p.begin(),p.end(),0);
        cnt=n;
    }

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

    void merge(int u,int v){
        int uu=find(u),vv=find(v);
        if(uu!=vv){
            if(s[uu]>s[vv]){
                p[vv]=uu;
                s[uu]+=s[vv];
            }else {
                p[uu]=vv;
                s[vv]+=s[uu];
            }
            cnt--;
        }
    }
    int size(int x){
        return s[x];
    }
};
相关推荐
虾球xz15 分钟前
游戏引擎学习第243天:异步纹理下载
c++·学习·游戏引擎
wolf犭良28 分钟前
37、aiomysql实操习题
开发语言·python·算法
xin007hoyo37 分钟前
算法笔记.spfa算法(bellman-ford算法的改进)
数据结构·笔记·算法
向哆哆39 分钟前
Java 加密与解密:从算法到应用的全面解析
java·开发语言·算法
uhakadotcom1 小时前
刚发布的PyTorch 2.7提供了什么 新特性
算法·面试·github
新生农民2 小时前
30分钟解决8道算法题
java·数据结构·算法
bbc1212262 小时前
2025/4/23 心得
数据结构·算法
Brookty3 小时前
【数据结构】Map与Set结构详解
数据结构
WaitWaitWait016 小时前
LeetCode每日一题4.20
算法·leetcode
蒟蒻小袁6 小时前
力扣面试150题--有效的括号和简化路径
算法·leetcode·面试