可撤销并查集板子

可撤销并查集是支持“撤销连接”操作的并查集,但是这个撤销必须按栈的顺序。

由于需要实现撤销,需要用栈记录所有已经完成过的连接操作。并且不能路径压缩。

板子如下:

#include <bits/stdc++.h>
using namespace std;
#define FOR(i, a, b) for (int i = (a); i <= (b); i++)
const int N = 2e5+5;
int n,fa[N],rk[N];
struct edge{int u,v,w;} stk[N]; int top = 0; //记录操作的栈,u和v是连接的两个点,w代表连接之前u和v是否等秩。top是栈顶

inline void init(){FOR(i,1,n) fa[i]=i, rk[i]=1;} //初始化
int find(int x){return fa[x]==x ? x : find(fa[x]);} //递归找祖先,注意这里不能路径压缩
bool same(int u,int v){return find(u)==find(v);} //判断u和v祖先是否相等
void merge(int u,int v){ //连接u和v
	u = find(u), v = find(v); //把u和v赋值为自身的祖先
	if(u==v) return; //如果祖先相同,不做任何操作
	if(rk[u] > rk[v]) swap(u,v); //按秩调整u和v的父子关系,小连大
	fa[u] = v; //连接
	stk[++top] = {u, v, rk[u]==rk[v]}; //把连接记录在栈中
	if(rk[u]==rk[v]) rk[v]++; //如果等秩,v的秩+1
}
void undone(){ //撤销上一次连接
	int u = stk[top].u, v = stk[top].v, w = stk[top].w;
	rk[v] -= w; //如果连接前等秩,那么现在v的秩-1
	fa[u] = u; //取消u,v之间的连接
	top--;
}

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
可持久化并查集是指在并查集的基础上,支持回退到任意历史版本。这个结构可以用来处理一些需要撤销或者回退操作的问题。以下是一个基本的可持久化并查集的实现。 ```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、付费专栏及课程。

余额充值