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

基础用法

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];
    }
};
相关推荐
艾醒32 分钟前
huggingface入门:如何使用国内镜像下载huggingface中的模型
算法
悠哉清闲35 分钟前
C ++代码学习笔记(一)
c++·笔记·学习
艾醒42 分钟前
huggingface入门:Tokenizer 核心参数与实战指南
算法
啊我不会诶1 小时前
【图论】拓扑排序
算法·深度优先·图论
浩浩乎@1 小时前
【openGLES】着色器语言(GLSL)
人工智能·算法·着色器
YxVoyager2 小时前
【C标准库】详解<stdio.h>标准输入输出库
c语言·c++
希望_睿智2 小时前
实战设计模式之解释器模式
c++·设计模式·架构
张同学的IT技术日记2 小时前
【奇妙的数据结构世界】 用经典例题对数组进行全面分析 | C++
算法
queenlll2 小时前
Codeforces Round 1043 (Div. 3)
算法
摸鱼一级选手2 小时前
十大经典 Java 算法解析与应用
java·算法·排序算法