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

基础用法

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];
    }
};
相关推荐
喝拿铁写前端23 分钟前
SmartField AI:让每个字段都找到归属!
前端·算法
莫有杯子的龙潭峡谷24 分钟前
3.31 代码随想录第三十一天打卡
c++·算法
LuckyLay44 分钟前
LeetCode算法题(Go语言实现)_22
算法·leetcode·golang
BingLin-Liu1 小时前
蓝桥杯备考---》贪心算法之矩阵消除游戏
算法·游戏·贪心算法
AaronZZH1 小时前
【进阶】vscode 中使用 cmake 编译调试 C++ 工程
c++·ide·vscode
我的sun&shine1 小时前
高级数据结构03RB树
数据结构·b树
杨筱毅2 小时前
【性能优化点滴】odygrd/quill 中将 MacroMetadata 变量声明为 constexpr
c++·性能优化
NaZiMeKiY2 小时前
C++ 结构体与函数
开发语言·c++
涛ing2 小时前
【Git “fetch“ 命令详解】
linux·c语言·c++·人工智能·git·vscode·svn
山河君2 小时前
音频进阶学习二十四——IIR滤波器设计方法
学习·算法·音视频·信号处理