并查集与路径压缩

1.1并查集的定义

所谓并查集就是“Union(合并)”,“Find(查找)”,“Set(集合)”对应的中文翻译,集合是指对集合操作的,而支持的操作就是Union和Find:
1)合并:合并两个集合。
2)查找:判断两个元素是否在一个集合。
好,定义我们知道了,那么并查集是如何实现的呢?其实就是一个数组:

int father[N];

其中father[i]表示元素i的父亲结点,而父亲结点本身也是这个集合内的元素(1<=i<=N)。例如father[1] = 2就表示元素的父亲结点是元素2,以这种父系关系来表示元素所属的集合。另外,如果father[i] = i,则说明元素i是该集合的根结点,且将其作为所属集合的标识。

1.2 并查集的基本操作

总体来说,并查集的使用需要初始化father数组,然后再根据需要进行查找或合并的操作。
1.2.1 初始化
一开始,每一个元素都是独立的一个集合,因此需要令所有father[i]等于i:

for(int i = 1;i<=N:i++){
	father[i] = i;//令father[i] = 1也可 这里只不过是拿father[i] = i举例 
} 

1.2.2 查找
由于规定同一个集合只存在一个根结点,因此查找操作就是对给定的结点寻找其根结点的过程。实现的方式可以是递推或递归,但是其思路都是一样的,即反复寻找父亲结点,直到根结点。(即father[i] = i的结点)。
递推写法如下:

//findFather函数返回元素x所在集合的结点
int findFather(int x){
	while(x!=father[x]){
		x = father[x];
	}
	return x;
} 

递归写法如下:

int findFather(int x){
	if(x == father[x])return x;
	else return findFather(father[x]);
}

1.2.3 合并
合并是指把两个集合合并成一个集合,题目中一般给出两个元素,要求把两个元素所在的集合合并。思路很清晰:
1)对于给定的两个元素a,b,判断它们是否属于同一个集合。可以调用findFather函数,对这个两个元素分别查找根结点,然后再判断根结点是否相同
2)合并两个集合:在1)中已经获得了两个元素的根结点faA与fbB,因此只需要把其中的一个父亲结点指向另一个结点。比如father[faA] = fbB / father[fbB] = faA都是可以的

void Union(int a,int b){
	int faA = findFather(a);
	int fbB = findFather(b);
	if(faA != fbB){
		father[faA] = fbB;//合并 
	}
}

这里提醒一下:很多初学者会直接把其中一个元素的父亲设为另一个元素,即直接令father[a] = b / father[b] = a来进行合并,这并不能实现将集合合并的效果。如果还是不懂,自己画图来理解。

1.3 路径压缩

事实上,上面讲解的并查集查找函数是没有经过优化的,在极端情况下效率较低。现在来考虑最极端的情况:题目给出的元素数量很多并且形成一条链,那么这个查找函数的效率会非常低。比如10^5个元素形成一条链,那么每次假设都要查找”最顶端的结点“(根结点),那每次的查找都有105的计算量,这显然不行。
好那该如何做呢,不妨来看个简单的例子:

father[1] = 1;
father[2] = 1;
father[3] = 2;
father[4] = 3;

因此如果只是为了查找跟结点的话,那么完全可以想办法把操作等价地变成:

father[1] = 1;
father[2] = 1;
fahter[3] = 1;
father[4] = 1;

可以自己画个图感受下变化。

这样相当于把当前查询结点的路径上的所有结点的父亲都指向根结点,查找的时候就不需要回溯去找父亲了,这样就能把极端情况降为O(1)了。
好,那么如何实现转换呢?可以回忆一下之前查找findFather()函数查找的过程,可以知道是从给定结点不断获得其父亲结点而最终到达根结点的。
因此转换的过程概括为以下两个步骤:
1)按原先的写法获得x的根结点r。
2)重新从x开始走一遍寻找根结点的过程,把路径上经过的所有结点的父亲全部改为根结点r。

可以写出代码:

int findFather(int x){
	int a = x;//由于x在下面的while中会变成根结点,因此先把原先的x保存一下 
	while(x != father[x]){
		x = father[x]; 
	}
	//到这里x存放的是根结点 下面把路径上的所有结点的father都改成根结点  
	while(a != father[a]){
		int z = a;//因为a要被father[a]覆盖 所以先保存a的值 以修改father[a]
		a = father[a];//a回溯父亲结点 
		father[z] = x;//返回根结点 
	} 
} 

这样就可以在查找时把寻找根结点的路径压缩了。
喜欢递归写法的人也可以像下面这样写:

int findFather(int v){
	if(v == father[v])return v;//找到根结点 
	else{
		int F = findFather(father[v]);//递归寻找father[v]的根结点F
		father[v] = F;
		return F; 
	}
} 
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值