当前位置:首页 > C++知识 > 正文内容

【数据结构】并查集1

亿万年的星光2年前 (2023-02-19)C++知识3673

1.引入

    对于一个集合S={a1, a2, …, an-1, an},我们还可以对集合S进一步划分: S1,S2,…,Sm-1,Sm,我们希望能够快速确定S中的两两元素是否属于S的同一子集。

    举个栗子,S={0,1, 2, 3, 4, 5, 6},如果我们按照一定的规则对集合S进行划分,假设划分后为S1={1, 2, 4}, S2={3, 6},S3={0, 5},任意给定两个元素,我们如何确定它们是否属于同一子集?某些合并子集后,又如何确定两两关系?基于此类问题便出现了并查集这种数据结构。

2.概念

    并查集是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。

并查集的思想是用一个数组表示了整片森林(parent),树的根节点唯一标识了一个集合,我们只要找到了某个元素的的树根,就能确定它在哪个集合里。

3.作用

    并查集可以高效的对元素进行分组(合并在一起),并且能快速的查询两个元素是否属于同一组。

4.基本操作

    合并:合并两个集合。join函数
    查找:判断两个元素是否在一个集合。find函数 

5.过程解释

    假如有两个集合,之间的关系可以用下面的图像表示:

一开始,分成两组,1和2是同一组,1和3是同一组,那么有没有什么办法可以将这两组练习起来。我们将1视为2的父级,1可以视为3的父级

这样,我们可以把图片划分成下面这样:

那么如果我们有其他的数据,比如下面的4,5,6这样表示

那么新问题来了,这两个集合如果在一块应该怎么表示呢?很简单,让这两个集合中的‘老大’打一架,谁赢了,谁就是两个集合共同的‘老大’。

假如 编号1 赢了,那么集合应该变成下面这样:

也就1成了所有人的‘老大’

那么也就是形成了一个等级严格的树形结构。

从数据结构上看,我们关注的重点是两个数据之间是否连通。

那么看下面这个例子

对于上面这个图,我们如何界定两个节点之间是否有关系呢?比如下面这两个题:
(1)问:2和3之间是否存在关系?

(2)问:3和7之间是否存在关系?

对于问题(1),我们从图中可以很明显的看出,2和3之间有关系,因为他们有共同的’老大‘。

对于问题(2),  我们从图中也可以明显的看出,3和7之间没有关系,因为从3开始往上找,发现老大是1,而从7开始找,我们发现老大是4,这两个老大(根节点)不一样,说明没有关系。

在做题中,也就是我们输入每组的对应关系如下,

1 2
1 3
4 5
4 6
6 7

然后询问任意两个数(节点)之间有没有关系?



6.代码实现

(1)初始化

我们用f数组表示父级元素节点,初始化代码如下

int f[100]; //父节点 

//初始化 n个元素 
void Init() {
	//使每个元素的根节点是其本身
	//即初始时每个元素都是单独的 
	for(int i=1; i<=n; i++){
			f[i]=i;  
	} 
}

’每个节点都是自己的父节点‘


(2)查询操作

查询操作有两种实现方式,一种是递归实现,另一种是非递归实现。

递归实现:

int find(int x){
    if(f[x] == x)
        return x;
    else
        return find(f[x]);}

一层一层访问父节点,直至根节点(根节点的标志就是父节点是本身)。要判断两个元素是否属于同一个集合,只需要看它们的根节点是否相同即可。

非递归实现:

int find(int x) {
	while(f[x] != x) {  //直到元素的父节点是它本身,表示已经查询到了树的根
		x= f[x];
	return x; 			//返回根节点对应的元素 
}



(3)合并

void merge(int a, int b) {
	//先找到两个元素对应的根对应的元素 
	int fa = find(a); 
	int fb = find(b);
	if(fa==fb) return;
	else f[fa]=fb;  //否则令元素 a的根指向元素 b的根 
}

合并操作也是很简单的,先找到两个集合的代表元素,然后将前者的父节点设为后者即可。当然也可以将后者的父节点设为前者。


6.例题

【题目描述】

如题,现在有一个并查集,你需要完成合并和查询操作。

【输入描述】

第一行包含两个整数n,m表示共有n个元素和m个操作。

接下来M行,每行3个整数z,x,y。

当z=1时,将x和y集合合并

当z=2时,判断x和y是否属于同一集合,如果是输出YES,否则输出NO

【输出描述】

对于每个z=2,判断输出YES或者NO

【样例输入】

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

【样例输出】

NO
YES
NO
YES

【参考代码】

#include<bits/stdc++.h>
using namespace std;
int n,m,f[10010];
int z,x,y;
int find(int x)//找出某棵树的老大 
{
	if(f[x]==x)//x是x的老大,
		return x;
	else 
		return find(f[x]);//x不是他自己的老大,所以他上面还有老大,递归 
}
void merge(int x,int y)
{
	//先找到两个元素对应的根对应的元素 
	int fx = find(x); 
	int fy = find(y);
	if(fx==fy) return;
	else f[fy]=fx;  //否则令元素 x的根指向元素 y的根  
	return ;
}
int main()
{
	cin>>n>>m; 
	//初始化每个人 
	for(int i=1;i<=n;i++)
		f[i]=i;

	//询问 
	for(int i=1;i<=m;i++)
	{
		cin>>z>>x>>y; 
		if(z==1){
			merge(x,y);
		}else{
			if(find(x)==find(y)){
				cout<<"YES"<<endl;
			}else{
				cout<<"NO"<<endl;
			} 
		}
		
	}
	return 0;
}


扫描二维码推送至手机访问。

版权声明:本文由青少年编程知识记录发布,如需转载请注明出处。

分享给朋友:

相关文章

C++中的max和min函数(最大值,最小值)

1.头文件      最大值最小值函数所在头文件是#include<algorithm>2.用法#include<iostream> #incl...

C++中的宏

一、预处理和编译器    首先,预编译器就是在编译器之前运行,换句话说,预编译器根据程序员的指示,决定实际要编译的内容。预编译器编译指令都以 # 开头。例如:1...

CSP-J2021年普及组复赛T4——小熊的果篮

【题目描述】    小熊的水果店里摆放着一排 n 个水果。每个水果只可能是苹果或桔子,从左到右依 次用正整数 1、2、3、……、n 编号。连续排在一起的同一种...

【STL】二分查找函数(算法)—binary_search

【说明】binary_search() 实现了一个二分查找算法。它会在前两个参数指定范围内搜索等同于第三个参数的元素。指定范围的迭代器必须是正向迭代器而且元素必须可以使用 < 运算符来比较。这个...

c++ 如何用链表存取数据

c++ 如何用链表存取数据

由于单链表的每个结点都有一个数据域和一个指针域。所以,每个结点可以定义成一个记录。其中,DATA数据元素,可以为你想要储存的任何数据格式,可以是数组,可以是int,甚至可以是结构体(这就是传说中的结构...

判断闰年

代码参考:#include<iostream>  using namespace std; //判断闰年的函数  int leap(...