并查集Union-find Sets

 并查集是主要用来解决元素分组的问题,只要出现给定一些元素组成一些不相交集合,然后给出几组某某元素之间存在关系,再询问某某元素是否是同一集合的问题,通常就需要并查集大显神威。

就比如判断两个人是否为亲戚关系。

 如果两个人人群之中相视一笑不知是不是远房表亲,此时就要扒关系了,你的爷爷辈是谁,我的爷爷辈是谁,一直找到祖先,才发现我们是远方表亲,于是把酒言欢。

 分析一下上边的一套操作,首先我们不知道两个元素是不是同一个集合中的,所以我们要觅根求源,我们的并查集就是这个作用,当然,前边是本来就有亲戚关系的,如果追溯到祖先都没有发现他们的关系,那他们就是茫茫人海相遇的陌生人罢了,当然,并查集是支持给两个元素搞上关系的。

 根据他的英文名就能知道,他是合并及查找合为一体的数据结构。

并查集支持两种操作:

  • 合并(Union):合并两个元素所属集合(合并对应的树)
  • 查询(Find):查询某个元素所属集合(查询对应的树的根节点),这可以用于判断两个元素是否属于同一集合

我们需要先进行初始化,然后再构建出他们的功能。

 起始,我们给出一组元素,我们可以用一个数组fa\[\]来存储每个结点的父节点,一开始,这几个元素没有任何关系,我们先将他们的父节点设为他们自己。

假设有0,1,2,3...n个元素。

cpp 复制代码
int fa[n];
void init(int n)
{
	for (int i = 0; i <= n; i++)
	{
		fa[i] = i;
	}
}

 要注意,数组中存放的是元素的祖先,如何进行合并操作呢?

我们先来看一看find操作。

cpp 复制代码
int find(int i)
{
	if (fa[i] == i)
		return i;
	else
		return find(fa[i]);
}

 find操作是查找该元素的祖先,如果该元素的祖先就是他自己,就返回他自己。

再来看一看合并操作。

cpp 复制代码
void union(int i, int j)
{
	int i_fa = find(i);
	int j_fa = find(j);
	fa[i_fa] = j_fa;
}

 如果单单看这两个函数,相信大家不会太明白到底是如何合并的。我们可以用几个示例来看一下。

 此时的话,数组就变为这样了,我们可以更加形象的表示数组和元素的对应关系。

 然后再合并2,3,此时3的祖先为4,2的祖先为2,就是将2的祖先设置为4。同样,如果合并1,2,就是将1的祖先设置为4。

此时

 如果此时查找是否1和5是同一集合,只需要查找他们各自的祖先判断是否相等即可。

但是上述find函数是有缺陷的。

如果我们这样合并呢?

 是不是还是觉得没有问题,如果我们再合并(2,1),合并后该结构就像一个链表一样。

此时如果我们合并(4,5)呢?

 我们要寻找4的祖先,此时数组中4位置为3,于是传递3,再次寻找3的祖先,还是不对,直至找到1,1的祖先还是1,然后将1的祖先置为5。

 如果继续合并(x,4),x是100时,我们就要向上递归100多次,我们不能直接找到合并数的祖先,需要递归查找好久,这样无疑很是浪费时间。

所以我们可以改进find函数

 路径压缩版本,在查找时,将路上的节点的祖先直接指向最终的祖先,而不是通过递归一步一步查找。

cpp 复制代码
int find(int i)
{
	if (i == fa[i])//查找到祖先还是进行返回
		return i;
	else {
		fa[i] = find(fa[i]);//接收返回值,并将最终的父节点赋值给路上的节点
	}
	return fa[i];//继续传递
}

此时再来进行一次模拟

然后继续递归向下寻找。

 返回值为1,此次递归结束后,返回调用该递归函数的位置,可以发现,调用2的位置会接收返回值,然后将2的祖先设置为该返回值。

 i等于2的递归函数结束后,返回到上一次的调用,即i等于3的位置。

然后继续返回上一层,将4的祖先也置为1。

这时,想要查找任意一个数字的祖先就可以很快查到。

 此时合并4,5,4的祖先为1,5的祖先为5,直接就可以将1的祖先设置为5,不像之前那样需要递归多次。

现在给出一个例题。
寻找图中是否存在路径

题目解析

 给出n个顶点,然后给出数组,数组中装的就是顶点和顶点之间的边,我们需要确定在根据数组中保存的边和边的关系,来推断某两个顶点是否相连。

就比如第一个例子

n等于3,就是有3个节点,分别为0,1,2。

 如果想要从0到2,我们可以通过1间接到达,也可以直接从0到2,所以返回结果为true。

看第二个例子

 正如上图,在将数组中的边连接后,会分成两个模块,但是不存在从0到达5的节点,所以此时返回false。

 这道题目就是并查集的经典题目,我们只需要将数组中给出的两两元素连接即可。

使用find和union函数就可以连接,在此之前记得初始化。

cpp 复制代码
class Solution {
public:
int find(int i)
{
	if (i == fa[i])//查找到祖先还是进行返回
		return i;
	else {
		fa[i] = find(fa[i]);//接收返回值,并将最终的父节点赋值给路上的节点
	}
	return fa[i];//继续传递
}
void Union(int i, int j)
{
	int i_fa = find(i);
	int j_fa = find(j);
	fa[i_fa] = j_fa;
}
int* fa=new int[1000000];
    bool validPath(int n, vector<vector<int>>& edges, int source, int destination) {
        //先初始化
        for(int i=0;i<n;i++)
        {
            fa[i]=i;
        }
        for(auto k:edges)
        {
            Union(k[0],k[1]);        
        }
        if(find(source)==find(destination))
        {
            return true;
        }
        return false;
    }
};
相关推荐
无限的鲜花2 分钟前
协程本质是函数加状态机——零基础深入浅出 C++20 协程
c++·算法·c++20
精明的身影11 分钟前
C++自学之路1:Hello world
开发语言·c++
旖-旎28 分钟前
《LeetCode 64 最小路径和 || LeetCode 174 地下城游戏》
c++·算法·leetcode·动态规划
森林古猿142 分钟前
再论斜率优化
c++·学习·算法
水利行业RTU手艺人1 小时前
RTU固件中的“数据保险箱”——离线数据补发系统设计
c++·stm32·单片机·物联网
心中有国也有家1 小时前
鸿蒙Flutter开发环境从零搭建教程(Windows/macOS双平台·避坑版)
学习·flutter·华为·harmonyos
gugucoding1 小时前
25. 【C语言】二进制文件与随机读写
c语言·开发语言
凯瑟琳.奥古斯特2 小时前
力扣1009补码解法C++实现
开发语言·c++·算法·leetcode·职场和发展
十月的皮皮3 小时前
C语言学习笔记20260706-栈的压入、弹出序列验证
c语言·笔记·学习
心中有国也有家3 小时前
Flutter 鸿蒙适配第一步:从 hive 迁移到 hive\_ce
hive·学习·flutter·华为·harmonyos