CF414E Mashmokh‘s Designed Problem 题解

Hypoc_

于 2021-03-30 11:34:30 发布

阅读量125

点赞数

分类专栏：题解_杂

本文链接：https://blog.csdn.net/a_forever_dream/article/details/115321068

版权

题解_杂专栏收录该内容

304 篇文章 6 订阅

订阅专栏

这篇博客主要介绍了如何使用欧拉序和Splay树来解决一棵以111为根的树的动态维护问题。通过Splay树以欧拉序为键，深度为值进行维护，支持求两点距离、子树迁移和深度查询等操作。文章详细解析了每个操作的实现思路，并给出了相应的C++代码实现，展示了在竞赛编程中解决复杂树形问题的方法。

摘要由CSDN通过智能技术生成

题目传送门

题目大意： 给一棵以 $1$ 为根的树，边有序号，维护三种操作：1、求两点距离；2、支持将一棵子树拔下来插到另一个节点上，成为他的最后一个儿子；3、从根出发dfs，求最后一个dfs到的深度为 $k$ 的节点。

题解

边有序号大概就是暗示你这棵树dfs序唯一，但是这题dfs序没有什么好的性质，但这时候我们灵机一动发现一个叫欧拉序的东西也是唯一的，拿过来用一用。

我们考虑用splay来维护，以欧拉序为key，深度为value。先看第二个操作，记 $in_x,out_x$ 分别为 $x$ 在欧拉序中第一次和第二次出现的位置，假如我们要将 $x$ 子树拔下来插到 $y$ 下，那么不难发现这个操作其实就是将 $in_x,out_x]$ 移到 $out_y$ 前，并且深度整体减去一个值，splay随便做做就好。

再看第三个操作，找到深度为 $k$ 且在dfs序最后面的点。这个点显然在欧拉序上也是在最后的，考虑在splay上维护子树中深度的 $\min$ 和 $\max$ ，然后优先走右儿子查找一下就好。

最后回到第一个操作，两点距离相当于找lca，不难发现lca是 $in_x,in_y)$ 这一段中的最小深度 $- 1$ （讨论一下 $x$ 是 $y$ 父亲和 $x$ 与 $y$ 不是父子关系两种情况就可以发现是正确的），那么就做完了。

~~不过话说赛时是什么神仙才写得出来这种东西啊QAQ……~~

代码如下：

#include <bits/stdc++.h>
using namespace std;
#define maxn 200010

int n,q;
vector<int> e[maxn];
struct node{
	int id,dep,mi,ma,lazy;node *zuo,*you,*fa;
	node(int Id,int x,node *Fa):id(Id),dep(x),mi(x),ma(x),lazy(0),zuo(NULL),you(NULL),fa(Fa){}
	void update(int x){lazy+=x;mi-=x;ma-=x;dep-=x;}
	void pushdown(){
		if(!lazy)return;
		if(zuo)zuo->update(lazy);
		if(you)you->update(lazy);
		lazy=0;
	}
	void check(){
		mi=ma=dep;
		if(zuo)mi=min(mi,zuo->mi),ma=max(ma,zuo->ma);
		if(you)mi=min(mi,you->mi),ma=max(ma,you->ma);
	}
	node *find(int x){
		pushdown();
		if(you&&you->mi<=x&&you->ma>=x)return you->find(x);
		if(dep==x)return this;
		return zuo->find(x);
	}
}*rt(NULL);
void rotate(node *x){
	node *fa=x->fa,*gfa=fa->fa;
	if(fa->zuo==x){
		fa->zuo=x->you;
		if(x->you)x->you->fa=fa;
		x->you=fa;
	}else{
		fa->you=x->zuo;
		if(x->zuo)x->zuo->fa=fa;
		x->zuo=fa;
	}
	fa->fa=x;x->fa=gfa;
	if(gfa&&gfa->zuo==fa)gfa->zuo=x;
	if(gfa&&gfa->you==fa)gfa->you=x;
	fa->check();x->check();
}
#define witch(x) (x->fa->zuo==x)
void splay(node *x,node *to=NULL){
	static node *sta[maxn];int t=0;
	sta[++t]=x;node *xx=x;
	while(xx->fa!=to)xx=sta[++t]=xx->fa;
	while(t)sta[t--]->pushdown();
	while(x->fa!=to){
		if(x->fa->fa!=to&&witch(x)==witch(x->fa))rotate(x->fa);
		rotate(x);
	}
	if(to==NULL)rt=x;
}
node *add(int Id,int dep){
	if(rt==NULL)return rt=new node(Id,dep,NULL);
	node *p=rt;while(p->you)p=p->you;
	p->you=new node(Id,dep,p);splay(p->you);
	return rt;
}
node *getpre(node *x){
	splay(x);x=x->zuo;
	while(x->you)x=x->you;
	return x;
}
node *getnxt(node *x){
	splay(x);x=x->you;
	while(x->zuo)x=x->zuo;
	return x;
}
void getseg(node *x,node *y){
	node *xx=getpre(x),*yy=getnxt(y);
	splay(xx);splay(yy,xx);
}
node *in[maxn],*out[maxn];
void dfs(int x,int dep){
	in[x]=add(x,dep);
	for(int y:e[x])dfs(y,dep+1);
	out[x]=add(x,dep);
}

int main()
{
	scanf("%d %d",&n,&q);
	for(int i=1;i<=n;i++){
		int k;scanf("%d",&k);
		int x;while(k--)scanf("%d",&x),e[i].push_back(x);
	}
	add(-1,-1);dfs(1,0);add(-1,-1);//在前后塞两个节点可以避免讨论边界问题
	for(int i=1,type,x,y;i<=q;i++){
		scanf("%d %d",&type,&x);
		if(type==1){
			scanf("%d",&y);
			if(x==y){puts("0");continue;}
			node *p1=in[x],*p2=in[y];
			//注意，取出(in[x],in[y])这个区间时要考虑此时两者欧拉序谁在前谁在后
			//如果你不小心忘掉了树会变化而使用一开始的欧拉序来判断，那么你就可以像我一样反复RE (
			splay(p1);splay(p2,p1);
			if(p1->zuo==p2)swap(p1,p2);

			getseg(getnxt(p1),p2);
			int lca=rt->you->zuo->mi-1;
			printf("%d\n",in[x]->dep+in[y]->dep-2*lca);
		}else if(type==2){
			scanf("%d",&y);getseg(in[1],in[x]);
			node *p=rt->you->zuo->find(in[x]->dep-y);p=out[p->id];

			getseg(in[x],out[x]);node *now=rt->you->zuo;now->update(y-1);
			if(now->fa->zuo==now)now->fa->zuo=NULL;else now->fa->you=NULL;
			now->fa->check();now->fa=NULL;

			getseg(p,p);p->pushdown();p->zuo=now;now->fa=p;splay(now);
		}else{
			printf("%d\n",rt->find(x)->id);
		}
	}
}