并查集(基础+带权以及可撤销并查集后期更新)

本文介绍了并查集数据结构,包括其基本概念(节点与根的关系)、查找根节点的方法(递归与路径压缩优化)、以及如何合并元素以判断连通性。并通过实例演示了如何使用并查集解决蓝桥幼儿园交友活动中的朋友关系判断问题。
摘要由CSDN通过智能技术生成

并查集

并查集是一种图形数据结构,用于存储图中结点的连通关系。

每个结点有一个父亲,可以理解为“一只伸出去的手”,会指向另一个点,初始时指向自己。一个点的根节点是该点的父亲的父亲的..的父亲,直到某个点的父亲是自己(根)。

当两个点的根相同时,我们就说他们时同一类,或者是连通的。

如下:7,5,1,3,6的根都是3,所以他们是连通的。

2,4是连通的,而2,6不连通,因为他们的根不同。

找根的方法

如果当前点不是根,就返回父亲的根。否则就是自己

用递归的方法实现

int find(int x)

{

        if(pre[x]==x)retrun x;

        return find(pre[x]);

}

并查集的合并

在并查集中所有的操作都在根上,假如我要使x和y两个点合并,我们只需要将find(x)指向find(y)或者find(y)指向find(x);

pre[find(x)]=find(y);

假如我们要合并4和6两点,我们只需要将2指向3或将3指向2.

路径压缩

找根函数的复杂度最坏情况下会达到O(n),如果查询次数较多的话效率将会非常低下。

我们可以在找根的过程中,将父亲指向根,从而实现路径压缩,这样可以使得找根的总体时间的复杂度为O(log n)。如下图,执行一次root(7)之后,沿途的点都会直接指向根3

int  find(int x){

        return pre[x]=(pre[x]==x?x:find(pre[x]));//当前的这个点是否是根,是根的话直接输出x不是根的话,去寻中这个根

}

例题

蓝桥幼儿园

题目描述

蓝桥幼儿园的学生是如此的天真无邪,以至于对他们来说,朋友的朋友就是自己的朋友。

小明是蓝桥幼儿园的老师,这天他决定为学生们举办一个交友活动,活动规则如下:

小明会用红绳连接两名学生,被连中的两个学生将成为朋友。

小明想让所有学生都互相成为朋友,但是蓝桥幼儿园的学生实在太多了,他无法用肉眼判断某两个学生是否为朋友。于是他起来了作为编程大师的你,请你帮忙写程序判断某两个学生是否为朋友(默认自己和自己也是朋友)。

输入描述

第 11 行包含两个正整数N,M,其中 N 表示蓝桥幼儿园的学生数量,学生的编号分别为1∼N。

之后的第2∼M+1 行每行输入三个整数op,x,y:

  • 如果 op=1,表示小明用红绳连接了学生 x 和学生 y 。
  • 如果 op=2,请你回答小明学生 x 和 学生 y 是否为朋友。

输出描述

对于每个op=2 的输入,如果 x 和 y 是朋友,则输出一行 YES,否则输出一行 NO

输入输出样例

示例 1

输入

5 5 
2 1 2
1 1 3
2 1 3
1 2 3 
2 1 2

输出

NO
YES
YES

代码 

package chsi;
import java.util.*;
public class chapter1 {
	static int []pre;//定义一个数组表示每个结点的根是指向谁的
	
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner scan=new Scanner(System.in);
		int n=scan.nextInt();
		int m=scan.nextInt();
		pre=new int[n];//初始化pre
		for(int i=0;i<n;i++) 
			pre[i]=i;//初始根都是指向它本身
		for(int i=0;i<m;i++) {
			int op=scan.nextInt();
			int x=scan.nextInt()-1;
			int y=scan.nextInt()-1;
			if(op==1) {
				union(x,y);
			}else {
				x=find(x);
				y=find(x);
				if(x==y) {
					System.out.println("YES");
				}
				else	System.out.println("No");
				
			}
		}
	}
	public static int find(int x) {
		if(pre[x]!=x) {
			pre[x]=find(pre[x]);
		}//表示当前结点不是我们的根节点
		return pre[x];
	}//先写一个find查询,路径压缩的方式
	public static void union(int x,int y) {
		x=find(x);
		y=find(y);
		if(x!=y) {
			pre[x]=y;
		}
		return;
	}
	
}

 

可持久化并查集是指在并查集基础上,支持回退到任意历史版本。这个结构可以用来处理一些需要撤销或者回退操作的问题。以下是一个基本的可持久化并查集的实现。 ```python class Node: def __init__(self, parent=None, rank=0): self.parent = parent self.rank = rank class PersistentUnionFind: def __init__(self, size): self.n = size self.roots = [None] * (2 * size) self.ranks = [None] * (2 * size) def make_set(self, v): self.roots[v] = Node(v) self.ranks[v] = 0 def find(self, node, version): if node.parent is None: return node if node.parent != node: node.parent = self.find(node.parent, version) return node.parent def union(self, x, y, version): x_root = self.find(self.roots[x], version) y_root = self.find(self.roots[y], version) if x_root == y_root: return False if self.ranks[x_root] < self.ranks[y_root]: x_root, y_root = y_root, x_root new_root = Node(x_root, self.ranks[x_root] + (self.ranks[x_root] == self.ranks[y_root])) self.roots[x] = self.roots[y] = new_root self.ranks[x_root] = self.ranks[y_root] = new_root.rank return True def get_version(self): return len(self.roots) // self.n - 1 def get_root(self, v, version): return self.find(self.roots[v], version).parent.val ``` 这个代码中,我们使用了一个 `Node` 类来表示每个节点,其中 `parent` 表示节点的父亲,`rank` 表示节点的秩。我们需要用一个 `roots` 数组来保存所有版本的根节点,以及一个 `ranks` 数组来保存所有节点的秩。`make_set` 函数用来初始化一个新节点,这个节点的父亲指向自己,秩为 0。`find` 函数用来找到节点所在的集合的根节点。如果节点的父亲不是根节点,那么我们就递归地寻找它的父亲。在递归返回之前,我们将所有遍历过的节点的父亲都更新为根节点,这样可以加速下次查找。`union` 函数用来将两个节点所在的集合合并。首先找到两个节点所在集合的根节点,如果根节点相同,那么这两个节点已经在同一个集合中,不需要再次合并。否则,我们将秩较小的根节点挂在秩较大的根节点下面,同时更新秩。`get_version` 函数用来获取当前版本号,而 `get_root` 函数则用来获取节点在指定版本中所在的集合的根节点。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值