并查集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;
    }
};
相关推荐
霁月风9 分钟前
设计模式——适配器模式
c++·适配器模式
jrrz082831 分钟前
LeetCode 热题100(七)【链表】(1)
数据结构·c++·算法·leetcode·链表
咖啡里的茶i1 小时前
Vehicle友元Date多态Sedan和Truck
c++
海绵波波1071 小时前
Webserver(4.9)本地套接字的通信
c++
@小博的博客1 小时前
C++初阶学习第十弹——深入讲解vector的迭代器失效
数据结构·c++·学习
南宫生2 小时前
贪心算法习题其四【力扣】【算法学习day.21】
学习·算法·leetcode·链表·贪心算法
爱吃喵的鲤鱼2 小时前
linux进程的状态之环境变量
linux·运维·服务器·开发语言·c++
懒惰才能让科技进步2 小时前
从零学习大模型(十二)-----基于梯度的重要性剪枝(Gradient-based Pruning)
人工智能·深度学习·学习·算法·chatgpt·transformer·剪枝
DARLING Zero two♡2 小时前
关于我、重生到500年前凭借C语言改变世界科技vlog.16——万字详解指针概念及技巧
c语言·开发语言·科技
7年老菜鸡2 小时前
策略模式(C++)三分钟读懂
c++·qt·策略模式