题目大意: 给一棵以 1 1 1 为根的树,边有序号,维护三种操作:1、求两点距离;2、支持将一棵子树拔下来插到另一个节点上,成为他的最后一个儿子;3、从根出发dfs,求最后一个dfs到的深度为 k k k 的节点。
题解
边有序号大概就是暗示你这棵树dfs序唯一,但是这题dfs序没有什么好的性质,但这时候我们灵机一动发现一个叫欧拉序的东西也是唯一的,拿过来用一用。
我们考虑用splay来维护,以欧拉序为key,深度为value。先看第二个操作,记 i n x , o u t x in_x,out_x inx,outx 分别为 x x x 在欧拉序中第一次和第二次出现的位置,假如我们要将 x x x 子树拔下来插到 y y y 下,那么不难发现这个操作其实就是将 [ i n x , o u t x ] [in_x,out_x] [inx,outx] 移到 o u t y out_y outy 前,并且深度整体减去一个值,splay随便做做就好。
再看第三个操作,找到深度为 k k k 且在dfs序最后面的点。这个点显然在欧拉序上也是在最后的,考虑在splay上维护子树中深度的 min \min min 和 max \max max,然后优先走右儿子查找一下就好。
最后回到第一个操作,两点距离相当于找lca,不难发现lca是 ( i n x , i n y ) (in_x,in_y) (inx,iny) 这一段中的最小深度 − 1 -1 −1(讨论一下 x x x 是 y y y 父亲和 x x x 与 y y 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);
}
}
}