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

【数据结构】并查集1

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

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;
}


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

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

分享给朋友:

相关文章

【初级篇】求最大公约数的方法

1.辗转相除法int gcd(int a,int b)  {       if(a%b==0...

【题解】围圈报数(约瑟夫问题)

【题解】围圈报数(约瑟夫问题)

【题目描述】有n个人依次围成一圈,从第1个人开始报数,数到第m个人出列,然后从出列的下一个人开始报数,数到第m个热呢又出列,... ,如此反复到所有的人全部出列为止。设n个人的编号分别为1,2,......

C++中的逻辑与运算

样例#include<iostream> using namespace std; int main(){ cout<<(1&1)...

2021CSP-J/S全国晋级二轮分数线公布

普及组CSP-J序号省市CSP-J人数CSP-J晋级晋级比例最高分晋级最低分1甘肃13413399.25%86152宁夏10310198.06%65243天津46345197.41%8615.54云南...

【题解】玩具

【题目描述】商店正在出售蒜头君最喜欢的系列玩具,在接下来的 " 周中,每周会出售其中的一款,同一款玩具不会重复出现。由于是蒜头君最喜欢的系列,他希望尽可能多地购买这些玩具,但是同一款玩具蒜头...

STL入门——容器3:map

一、定义    Map是STL的一个关联容器,它提供一对一(其中第一个可以称为关键字,每个关键字只能在map中出现一次,第二个可能称为该关键字的值)的数据&nb...