可持久化并查集

题目

前置知识:主席树,按秩合并并查集

首先这题不能用压缩路径并查集,我也没搞清楚,所以使用按秩合并。

按秩合并

我更喜欢叫按深度合并,就是深度小的链接到深度大的树上,至于为什么,如图,如果反过来,我们的最大深度会继续加深,若干次操作后深度会很巨大,每次遍历都要跑很多层,然而小并大就能有效减少层数

 为了实现这个操作,我们可以定义寻找父节点的函数,若当前的节点父节点不为自己就递归调用,为此,我们有可以定义函数,根据编号找到某节点在树的哪个位置代码如下

int sfind(int x,int l,int r,int s)
{
	if(l==r) return x;
	int mid=(l+r)/2;
	if(s<=mid) return sfind(tree[x].l,l,mid,s);
	else return sfind(tree[x].r,mid+1,r,s);
}
int que(int x,int s)
{
	int tem=sfind(x,1,m,s);
	if(tree[tem].fa==s) return tem;
	return que(x,tree[tem].fa);
}

我们就完成了寻找父节点的部分

合并节点部分,我们需要比较深度,然后可持久化,注意如果深度一样的话就需要再将深度再加一,同时这里如果在加一操作时不新建版本的话会跑很久(我也不清楚为什么

代码如下

int x=que(root[i],b),y=que(root[i],c);
			if(tree[x].fa!=tree[y].fa)
			{
				if(tree[x].dep>tree[y].dep) swap(x,y);
				root[i]=com(root[i],1,m,tree[x].fa,tree[y].fa);
				if(tree[x].dep==tree[y].dep)
				root[i]=up(root[i],1,m,tree[y].fa);
			}
int com(int x,int l,int r,int a,int b)
{
	int n=t++;
	tree[n]=tree[x];
	if(l==r)
	{
		tree[n].fa=b;
		return n;
	}
	int mid=(l+r)/2;
	if(a<=mid) tree[n].l=com(tree[n].l,l,mid,a,b);
	else tree[n].r=com(tree[n].r,mid+1,r,a,b);
	return n;
}
int up(int x,int l,int r,int s)
{
	int n=t++;
	tree[n]=tree[x];
	if(l==r)
	{
		tree[n].dep++;
		return n;
	}
	int mid=(l+r)/2;
	if(s<=mid) tree[n].l=up(tree[n].l,l,mid,s);
	else tree[n].r=up(tree[n].r,mid+1,r,s);
	return n;
}

最后此题就剩一些输入输出细节了

完整代码

#include <bits/stdc++.h>
using namespace std;
#define N 300000
typedef struct node
{
	int l,r,fa,dep;
}node;
node tree[N*30];
int root[N*30];
int t=0,m,q;
int build(int l,int r)//建树
{
	int n=t++;
	if(l==r)
	{
		tree[n].fa=l;
		tree[n].dep=0;
		return n;
	}
	int mid=(l+r)/2;
	tree[n].l=build(l,mid);
	tree[n].r=build(mid+1,r);
	return n;
}
int sfind(int x,int l,int r,int s)//节点标号,范围,编号
{
	if(l==r) return x;
	int mid=(l+r)/2;
	if(s<=mid) return sfind(tree[x].l,l,mid,s);
	else return sfind(tree[x].r,mid+1,r,s);
}
int que(int x,int s)//节点标号,编号//递归寻找父节点
{
	int tem=sfind(x,1,m,s);
	if(tree[tem].fa==s) return tem;
	return que(x,tree[tem].fa);
}
int com(int x,int l,int r,int a,int b)//节点标号,范围,两个祖先//建新版本
{
	int n=t++;
	tree[n]=tree[x];
	if(l==r)
	{
		tree[n].fa=b;
		return n;
	}
	int mid=(l+r)/2;
	if(a<=mid) tree[n].l=com(tree[n].l,l,mid,a,b);
	else tree[n].r=com(tree[n].r,mid+1,r,a,b);
	return n;
}
int up(int x,int l,int r,int s)//节点标号,范围,编号//建新版本并加深度
{
	int n=t++;
	tree[n]=tree[x];
	if(l==r)
	{
		tree[n].dep++;
		return n;
	}
	int mid=(l+r)/2;
	if(s<=mid) tree[n].l=up(tree[n].l,l,mid,s);
	else tree[n].r=up(tree[n].r,mid+1,r,s);
	return n;
}

int main()
{
	scanf("%d%d",&m,&q);
	root[0]=build(1,m);
	for(int i=1;i<=q;i++)
	{
		int a,b,c;
		scanf("%d",&a);
		if(a==1)
		{
			scanf("%d%d",&b,&c);
			root[i]=root[i-1];
			int x=que(root[i],b),y=que(root[i],c);
			if(tree[x].fa!=tree[y].fa)
			{
				if(tree[x].dep>tree[y].dep) swap(x,y);//深度小并到深度大
				root[i]=com(root[i],1,m,tree[x].fa,tree[y].fa);
				if(tree[x].dep==tree[y].dep)
				root[i]=up(root[i],1,m,tree[y].fa);
			}
		}
		else if(a==2)
		{
			scanf("%d",&b);
			root[i]=root[b];
		}
		else if(a==3)
		{
			scanf("%d%d",&b,&c);
			root[i]=root[i-1];
			int x=que(root[i],b),y=que(root[i],c);
			if(tree[x].fa==tree[y].fa) printf("1\n");
			else printf("0\n");
		}
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 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、付费专栏及课程。

余额充值