数据结构-- 并查集

0. 引入

并查集是来解决等价问题的数据结构。

离散数学中的二元关系。

等价关系需满足自反性、对称性、传递性。
a ∈ S , a R a a R b & b R a a R b ∩ b R c = > a R c a \in S, aRa \\ aRb \& bRa \\ aRb \cap bRc =>aRc a∈S,aRaaRb&bRaaRb∩bRc=>aRc

1. 需要实现的操作

给定n个数据,看能划分多少个等价类。

初始时即分为n个等价类,然后再一一合并。

所以需要实现的操作为:

  1. 合并两个等价类
  2. 查找元素属于哪个等价类

2. 实现

2.0 父节点
cpp 复制代码
vector<int> pa;
2.1 查找
cpp 复制代码
int Find(int k)
{
	return k == pa[k] ? k : Find(pa[k]);
}
2.2 合并
cpp 复制代码
void Union(int a0, int a1)
{
	int p0 = Find(a0);
	int p1 = Find(a1);
	
	if ( p0 != p1 ) {
		pa[p0] = p1;
	}
}
2.3 路径压缩

对于查找来说如果简单的递归的话,最坏的情况便是全都在左子树。

(0,1) (0,2) (0,3) (0, 4) ... (0, n)

这样会导致单次查询如同一个链表一样达到O(n)

只需要改动一点点就可以完成路径压缩。

cpp 复制代码
int Find(int k)
{
return k == pa[k] ? k : pa[k] = Find(pa[k]);
}
2.4 按节点数合并

可以令开一个数组,记录当前节点下的节点数。在合并的时候取小的节点合并到大的节点上去。

cpp 复制代码
void Union(int a1, int a2)
{
	int p1 = Find(a1);
	int p2 = Find(a2);
	if ( p1 == p2)
	       return;
	
	if (sz[p1] < sz[p2]) {
		pa[p1] = p2;
	    sz[p2] += sz[p1];
	}
	else {
        pa[p2] = p1;
        sz[p1] += sz[p2];
    }     
}

3. 类封装

3.1 路径压缩
cpp 复制代码
class UnionFind {
    public:
        explicit UnionFind(int sz):cnt(sz),pa(sz)
        {
            iota(pa.begin(), pa.end(), 0);
        }
        int Find(int k )
        {
            return k == pa[k] ? k : pa[k] = Find(pa[k]);
        }
        void Union(int k1, int k2 )
        {
            int p0 = Find(k1);
            int p1 = Find(k2);

            if ( p0 != p1) {
                pa[p0] = p1;
                cnt--;
            }
        }
        int Cnt()
        {return cnt;}

    private:
        vector<int> pa;
        int cnt;
};
3.2 按节点数合并
cpp 复制代码
public:
class UnionFind {
    public:
        explicit UnionFind(int _sz):cnt(_sz),pa(_sz),sz(_sz, 1)
        {
            iota(pa.begin(), pa.end(), 0);
        }
        int Find(int k )
        {
            return k == pa[k] ? k : Find(pa[k]);
        }
        void Union(int k1, int k2 )
        {
            int p0 = Find(k1);
            int p1 = Find(k2);

            if (p0 == p1)
                return ;

            if (sz[p0] < sz[p1] ) {
                pa[p0] = p1;
                sz[p1] += sz[p0];
            }
            else {
                pa[p1] = p0;
                sz[p0] += sz[p1];
            }
        }
        int Cnt()
        {return cnt;}

        int Size(int idx)
        { return sz[idx]; }

        

    private:
        vector<int> pa,sz;
        int cnt;
};
4. 参考

lFoll题解
OIWIKI

相关推荐
passer__jw76710 分钟前
【LeetCode】【算法】283. 移动零
数据结构·算法·leetcode
Ocean☾16 分钟前
前端基础-html-注册界面
前端·算法·html
顶呱呱程序24 分钟前
2-143 基于matlab-GUI的脉冲响应不变法实现音频滤波功能
算法·matlab·音视频·matlab-gui·音频滤波·脉冲响应不变法
爱吃生蚝的于勒1 小时前
深入学习指针(5)!!!!!!!!!!!!!!!
c语言·开发语言·数据结构·学习·计算机网络·算法
羊小猪~~1 小时前
数据结构C语言描述2(图文结合)--有头单链表,无头单链表(两种方法),链表反转、有序链表构建、排序等操作,考研可看
c语言·数据结构·c++·考研·算法·链表·visual studio
王哈哈^_^1 小时前
【数据集】【YOLO】【VOC】目标检测数据集,查找数据集,yolo目标检测算法详细实战训练步骤!
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·pyqt
星沁城1 小时前
240. 搜索二维矩阵 II
java·线性代数·算法·leetcode·矩阵
脉牛杂德2 小时前
多项式加法——C语言
数据结构·c++·算法
legend_jz2 小时前
STL--哈希
c++·算法·哈希算法
kingmax542120082 小时前
初三数学,最优解问题
算法