union find算法 c++

1.原理参考

labuladong-fucking-algorithm/算法思维系列/UnionFind算法详解.md at master · jiajunhua/labuladong-fucking-algorithm · GitHub

2.初级模式

cpp 复制代码
#include <iostream>

class UF {

    public:
    // 记录连通分量

        /* 构造函数,n 为图的节点总数 */
        UF(int n) {
        count = n;
        //    int arr[n];
        parent_arr = new int[n]; 
        for(int i=0; i<n; i++)
        {
            parent_arr[i] = i;
        }
        };
        /* 其他函数 */

        ~UF()
        {
            delete[] parent_arr;
        }

        void union_func(int p, int q);

        int find(int x);

        int getCount();

        bool connect(int p, int q);

    private:
        int count;
        // 节点 x 的节点是 parent[x]
        //  int[] parent;
        int* parent_arr; 

};

void UF::union_func(int p, int q)
{
    int rootP = find(p);
    int rootQ = find(q);

    if(rootQ == rootP)
        return;
    
    parent_arr[rootQ] = rootP;

    count--;

}


int UF::find(int x)
{
    while (parent_arr[x] != x)
    {
        x = parent_arr[x];
    }

    return x;
}


int UF::getCount()
{
    return count;
}


bool UF::connect(int p, int q)
{
    int rootP = find(p);
    int rootQ = find(q);

    return rootP == rootQ;
}


int main()
{
    UF union_find(7);

    union_find.union_func(0, 1);
    union_find.union_func(0, 2);
    union_find.union_func(0, 3);
    union_find.union_func(2, 4);
    union_find.union_func(2, 5);
    union_find.union_func(3, 6);

    std::cout << union_find.getCount() << std::endl;

    return 0;
}
  1. 进阶模式

3.1 平衡性优化

3.2 路径压缩

cpp 复制代码
#include <iostream>

class UF {

    public:
    // 记录连通分量

        /* 构造函数,n 为图的节点总数 */
        UF(int n) {
        count = n;
        //    int arr[n];
        parent_arr = new int[n]; 
        size_arr = new int[n];
        for(int i=0; i<n; i++)
        {
            parent_arr[i] = i;
            size_arr[i] = i;
        }
        };
        /* 其他函数 */

        ~UF()
        {
            delete[] parent_arr;
        }

        void union_func(int p, int q);

        int find(int x);

        int getCount();

        bool connect(int p, int q);

    private:
        int count;
        int* parent_arr;
        int* size_arr; 

};

void UF::union_func(int p, int q)
{
    int rootP = find(p);
    int rootQ = find(q);

    // 小树接到大树下面,较平衡
    if (size_arr[rootP] > size_arr[rootQ]) {
        parent_arr[rootQ] = rootP;
        size_arr[rootP] += size_arr[rootQ];
    } else {
        parent_arr[rootP] = rootQ;
        size_arr[rootQ] += size_arr[rootP];
    }

    count--;

}


int UF::find(int x)
{
    while (parent_arr[x] != x)
    {
        // 进行路径压缩
        parent_arr[x] = parent_arr[parent_arr[x]];
        x = parent_arr[x];
    }

    return x;
}


int UF::getCount()
{
    return count;
}


bool UF::connect(int p, int q)
{
    int rootP = find(p);
    int rootQ = find(q);

    return rootP == rootQ;
}


int main()
{
    UF union_find(7);

    union_find.union_func(0, 1);
    union_find.union_func(0, 2);
    union_find.union_func(0, 3);
    union_find.union_func(2, 4);
    union_find.union_func(2, 5);
    union_find.union_func(3, 6);

    std::cout << union_find.getCount() << std::endl;

    return 0;
}
相关推荐
Mercury-circle1 分钟前
JavaScript基础知识合集笔记1——数据类型
开发语言·javascript·笔记
Small踢倒coffee_氕氘氚8 分钟前
是否应该禁止危险运动论文
经验分享·笔记·算法·灌灌灌灌
Chase_______10 分钟前
Java后端开发——分层解耦详解
java·开发语言·spring·web
喝可乐的布偶猫12 分钟前
Java----super 关键字
java·开发语言
篱笆院的狗19 分钟前
Java 中 ConcurrentHashMap 1.7 和 1.8 之间有哪些区别?
java·开发语言
Maple_land43 分钟前
C++初阶——string的模拟实现(上)
c++
今晚打老虎1 小时前
c++弹窗
开发语言·c++
虾球xz1 小时前
游戏引擎学习第240天:将渲染器移至第三层
c++·学习·游戏引擎
java1234_小锋1 小时前
什么是Lua模块?你会如何使用NGINX的Lua模块来定制请求处理流程?
开发语言·nginx·lua
web守墓人1 小时前
【go语言】window环境从源码编译go
开发语言·后端·golang