2018.10.09 bzoj4817: [Sdoi2017]树点涂色（lct）

传送门
其实这道题跟真正的lct关系不是很大。
首先观察到操作1和$access$操作很像，启示我们每次把相同颜色的看成一棵$splay$，那么最后路径$u$->$v$的权值就是路径上$splay$的数量。
这样的话我们每个点$i$维护一个$1$->$i$的权值$w_i$。
分类讨论一波之后会发现无论是什么情况路径$u$->$v$的权值总是等于$w_u+w_v-2\ast w_{lca}+1$。
而对于操作3直接线段树+$dfs$序维护子树最值就行了。
代码：

#include<bits/stdc++.h>
#define N 100005
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans; 
}
int n,m,first[N],siz[N],st[N][21],pred[N],dfn[N],dep[N],cnt=0,tot=0;
struct edge{int v,next;}e[N<<1];
struct Node{int l,r,mx,add;};
inline int max(int a,int b){return a>b?a:b;}
struct sgt{
	Node T[N<<2];
	#define lc (p<<1)
	#define rc (p<<1|1)
	#define mid (T[p].l+T[p].r>>1)
	inline void pushup(int p){T[p].mx=max(T[lc].mx,T[rc].mx);}
	inline void pushnow(int p,int v){T[p].mx+=v,T[p].add+=v;}
	inline void pushdown(int p){if(T[p].add)pushnow(lc,T[p].add),pushnow(rc,T[p].add),T[p].add=0;}
	inline void build(int p,int l,int r){
		T[p].l=l,T[p].r=r;
		if(l==r){T[p].mx=dep[pred[l]];return;}
		build(lc,l,mid),build(rc,mid+1,r),pushup(p);
	}
	inline void update(int p,int ql,int qr,int v){
		if(ql<=T[p].l&&T[p].r<=qr)return pushnow(p,v);
		pushdown(p);
		if(qr<=mid)update(lc,ql,qr,v);
		else if(ql>mid)update(rc,ql,qr,v);
		else update(lc,ql,mid,v),update(rc,mid+1,qr,v);
		pushup(p);
	}
	inline int query(int p,int ql,int qr){
		if(ql<=T[p].l&&T[p].r<=qr)return T[p].mx;
		pushdown(p);
		if(qr<=mid)return query(lc,ql,qr);
		if(ql>mid)return query(rc,ql,qr);
		return max(query(lc,ql,mid),query(rc,mid+1,qr)); 
	}
}sgt;
struct link_cut_tree{
	int son[N][2],fa[N];
	inline int which(int x){return x==son[fa[x]][1];}
	inline bool isroot(int x){return !fa[x]||(x!=son[fa[x]][0]&&x!=son[fa[x]][1]);}
	inline void rotate(int x){
		int y=fa[x],z=fa[y],t=which(x);
		if(z&&!isroot(y))son[z][which(y)]=x;
		fa[y]=x,fa[x]=z,son[y][t]=son[x][t^1],son[x][t^1]=y;
		if(son[y][t])fa[son[y][t]]=y;
	}
	inline void splay(int x){while(!isroot(x)){if(!isroot(fa[x]))rotate(which(x)==which(fa[x])?fa[x]:x);rotate(x);}}
	inline int find(int x){while(son[x][0])x=son[x][0];return x;}
	inline void access(int x){
		for(int y=0;x;y=x,x=fa[x]){
			splay(x);
			if(son[x][1]){
				int tmp=find(son[x][1]);
				sgt.update(1,dfn[tmp],dfn[tmp]+siz[tmp]-1,1);
			}
			son[x][1]=y;
			if(son[x][1]){
				int tmp=find(son[x][1]);
				sgt.update(1,dfn[tmp],dfn[tmp]+siz[tmp]-1,-1);
			}
		}
	}
}lct;
inline void dfs(int p){
	dep[p]=dep[st[p][0]]+1,pred[dfn[p]=++tot]=p,siz[p]=1;
	for(int i=1;i<=20;++i)st[p][i]=st[st[p][i-1]][i-1];
	for(int i=first[p];i;i=e[i].next){
		int v=e[i].v;
		if(v==st[p][0])continue;
		st[v][0]=p,lct.fa[v]=p,dfs(v),siz[p]+=siz[v];
	}
}
inline int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	int tmp=dep[x]-dep[y];
	for(int i=20;~i;--i)if((tmp>>i)&1)x=st[x][i];
	if(x==y)return x;
	for(int i=20;~i;--i)if(st[x][i]!=st[y][i])x=st[x][i],y=st[y][i];
	return st[x][0];
}
inline void add(int u,int v){e[++cnt].v=v,e[cnt].next=first[u],first[u]=cnt;}
int main(){
	n=read(),m=read();
	for(int i=1,u,v;i<n;++i)u=read(),v=read(),add(u,v),add(v,u);
	dfs(1),sgt.build(1,1,n);
	while(m--){
		int op=read(),x=read(),y,z;
		if(op==1)lct.access(x);
		else if(op==2)y=read(),z=lca(x,y),printf("%d\n",sgt.query(1,dfn[x],dfn[x])+sgt.query(1,dfn[y],dfn[y])-2*sgt.query(1,dfn[z],dfn[z])+1);
		else printf("%d\n",sgt.query(1,dfn[x],dfn[x]+siz[x]-1));
	}
	return 0;
}