并查集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;
    }
};
相关推荐
南东山人13 分钟前
一文说清:C和C++混合编程
c语言·c++
stm 学习ing1 小时前
FPGA 第十讲 避免latch的产生
c语言·开发语言·单片机·嵌入式硬件·fpga开发·fpga
Red Red2 小时前
网安基础知识|IDS入侵检测系统|IPS入侵防御系统|堡垒机|VPN|EDR|CC防御|云安全-VDC/VPC|安全服务
网络·笔记·学习·安全·web安全
Ysjt | 深3 小时前
C++多线程编程入门教程(优质版)
java·开发语言·jvm·c++
ephemerals__3 小时前
【c++丨STL】list模拟实现(附源码)
开发语言·c++·list
Microsoft Word3 小时前
c++基础语法
开发语言·c++·算法
Natural_yz3 小时前
大数据学习17之Spark-Core
大数据·学习·spark
qq_172805593 小时前
RUST学习教程-安装教程
开发语言·学习·rust·安装
一只小小汤圆4 小时前
opencascade源码学习之BRepOffsetAPI包 -BRepOffsetAPI_DraftAngle
c++·学习·opencascade