并查集Union-find Sets

简介: 并查集Union-find Sets


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

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

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

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

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

并查集支持两种操作:

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

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

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

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

int fa[n];
void init(int n)
{
  for (int i = 0; i <= n; i++)
  {
    fa[i] = i;
  }
}

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

我们先来看一看find操作。

int find(int i)
{
  if (fa[i] == i)
    return i;
  else
    return find(fa[i]);
}

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

再来看一看合并操作。

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函数

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

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函数就可以连接,在此之前记得初始化。

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;
    }
};
目录
相关文章
|
6月前
|
存储 机器学习/深度学习 算法
并查集(Union Find)
并查集(Union Find)
61 3
|
机器人 Python
并查集(Union-Find)
并查集(Union-Find)是一种用于解决动态连通性问题的数据结构,它主要用于处理不相交的集合(Disjoint Sets)之间的合并和查询操作。并查集的主要优点是,它不需要比较相邻的元素,而是通过分配和收集元素来进行操作,从而在处理大量数据时非常高效。
70 8
LeetCode 389. Find the Difference
给定两个字符串 s 和 t,它们只包含小写字母。 字符串 t 由字符串 s 随机重排,然后在随机位置添加一个字母。 请找出在 t 中被添加的字母。
118 0
LeetCode 389. Find the Difference
|
算法 Python
LeetCode 108. 将有序数组转换为二叉搜索树 Convert Sorted Array to Binary Search Tree
LeetCode 108. 将有序数组转换为二叉搜索树 Convert Sorted Array to Binary Search Tree
|
算法 容器
常用查找算法 find() find_if() adjacent_find() binary_search() count() count_if()
常用查找算法 find() find_if() adjacent_find() binary_search() count() count_if()
|
算法 容器
常用集合算法 set_intersection() set_union() set_difference()
常用集合算法 set_intersection() set_union() set_difference()
Data Structures and Algorithms (English) - 6-10 Sort Three Distinct Keys(30 分)
Data Structures and Algorithms (English) - 6-10 Sort Three Distinct Keys(30 分)
103 0
|
C++
Data Structures and Algorithms (English) - 6-9 Sort Three Distinct Keys(20 分)
Data Structures and Algorithms (English) - 6-9 Sort Three Distinct Keys(20 分)
112 0
LeetCode之Find the Difference
LeetCode之Find the Difference
118 0
|
算法 JavaScript Java
并查集算法 - Algorithms, Part I, week 1 UNION-FIND
cousera 普林斯顿大学 算法公开课 第一周 并查集数据类型内容整理
1506 0
下一篇
无影云桌面