[CF414E][平衡树][欧拉序]Mashmokh's Designed Problem

CF414E

dfs序上的深度是连续变化的
所以可以维护区间深度最大和最小的点
查询一个点最近的深度小于它的点就是fa
两个点之间深度最小的点就是lca
都可以用平衡树维护
对于第三个询问,直接找出最右边的深度为k的点即可

Code:

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=2e5+5,INF=1e9; 
namespace Splay{
	int ls[N],rs[N],fa[N];
	inline int isrs(int x){return rs[fa[x]]==x;}
	inline bool isroot(int x){
		if(!fa[x]) return 1;
		return ls[fa[x]]!=x && rs[fa[x]]!=x;
	}
	int s[N],si[N],mn[N],mx[N];
	inline void pushup(int x){
		s[x]=s[ls[x]]+s[rs[x]]+si[x];
		mx[x]=max(mx[ls[x]],s[ls[x]]+si[x]+max(mx[rs[x]],0));
		mn[x]=min(mn[ls[x]],s[ls[x]]+si[x]+min(mn[rs[x]],0));
	}
	inline void rotate(int x){
		int y=fa[x],z=fa[y],b=(ls[y]==x?rs[x]:ls[x]);
		if(z && !isroot(y)) (ls[z]==y?ls[z]:rs[z])=x;
		fa[x]=z,fa[y]=x,b?fa[b]=y:0;
		if(ls[y]==x) rs[x]=y,ls[y]=b;
		else ls[x]=y,rs[y]=b;
		pushup(y);pushup(x);
	}
	inline void splay(int x,int f){
		while(fa[x]!=f){
			if(fa[fa[x]]!=f){
				if(isrs(x)==isrs(fa[x])) rotate(fa[x]);
				else rotate(x);
			}
			rotate(x);
		}
	}
	int rt;
	int ff[N];
	inline int query(int x,int y){
		x=(x<<1)-1,y=(y<<1)-1;
		int xx,yy,lca;
		splay(x,0);xx=s[ls[x]]+si[x];
		splay(y,0);yy=s[ls[y]]+si[y];
		splay(x,0);splay(y,x);
		rt=x;
		lca=min(xx,yy);
		if(ls[x]==y) lca=min(lca,s[ls[y]]+si[y]+mn[rs[y]]);
		else lca=min(lca,s[ls[x]]+si[x]+mn[ls[y]]);
		return xx+yy-(lca<<1);
	}
	int find(int x,int rk){
		int k=rk-s[ls[x]]-si[x];
		if(mx[rs[x]]>=k && mn[rs[x]]<=k) return find(rs[x],k);
		if(s[ls[x]]+si[x]==rk) return (x&1)?((x+1)>>1):ff[x>>1];
		return find(ls[x],rk); 
	}
	inline int Pre(int x){splay(x,0);x=ls[x];while(rs[x]) x=rs[x];return x;}
	inline int Nxt(int x){splay(x,0);x=rs[x];while(ls[x]) x=ls[x];return x;}
}
using namespace Splay;
namespace Graph{
	int dfn[N],sign=0;
	int vis[N<<1],nxt[N<<1],head[N],tot=0;
	inline void add(int x,int y){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;}
	inline void dfs(int v){
		dfn[++sign]=(v<<1)-1;
		si[(v<<1)-1]=1;
		for(int i=head[v];i;i=nxt[i]){ff[vis[i]]=v;dfs(vis[i]);}
		dfn[++sign]=(v<<1);
		si[v<<1]=-1;
	}
	inline int build(int l,int r){
		int mid=l+r>>1;
		int &x=dfn[mid];
		if(l<mid) ls[x]=build(l,mid-1),fa[ls[x]]=x;
		if(mid<r) rs[x]=build(mid+1,r),fa[rs[x]]=x;
		pushup(x);return x;
	}
}
using namespace Graph;
inline void modify(int v,int nw){
	int x=(v<<1)-1,l,r,t;
	splay(x,0);
	ff[v]=find(ls[x],s[ls[x]]+si[x]-nw);
	l=Pre(x);r=Nxt(v<<1);
	splay(l,0);splay(r,l);
	t=ls[r];ls[r]=0;
	pushup(r);pushup(l);
	l=Pre(ff[v]<<1),r=(ff[v]<<1);
	splay(l,0);splay(r,l);
	ls[r]=t;fa[t]=r;
	pushup(r);pushup(l);
	rt=l; 
}
int tmp[N];
int main(){
	mx[0]=-INF,mn[0]=INF;
	int n=read(),m=read();
	for(int i=1;i<=n;i++){
		int num=read();
		for(int j=1;j<=num;j++) tmp[j]=read();
		for(int j=num;j;j--) add(i,tmp[j]);
	}
	dfs(1);rt=build(1,n<<1);
	while(m--){
		int op=read(),x=read();
		if(op==1){int y=read();cout<<query(x,y)<<"\n";}
		if(op==2){int y=read();modify(x,y);}
		if(op==3) cout<<find(rt,x+1)<<"\n";
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值