2019.03.21【SPOJ-QTREE5】Query on a tree V（LCT）

传送门

---

解析：

维护的东西和QTREE4大同小异。但是由于不用维护全局答案且边不带权，比QTREE4好写到不知道哪里去了。

开一个multiset记录从每个虚子树叉出去的最近距离。

用siz表示当前链长，$lmn$表示从链顶叉出来的最小距离，$rmn$表示从链底叉出去的最小距离，于是就可以非常愉快的转移了。

询问的时候直接access将当前点变成链底然后用$rmn$回答就行了。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<20|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++; 
	}
	
	inline int getint(){
		re char c;
		while(!isdigit(c=gc()));re int num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
}
using namespace IO;

using std::cout;

inline int min(cs int &a,cs int &b){return a<b?a:b;}

cs int N=1e5+5;
int n,q;
int last[N],nxt[N<<1],to[N<<1],ecnt;
inline void addedge(int u,int v){
	nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v;
	nxt[++ecnt]=last[v],last[v]=ecnt,to[ecnt]=u;
}

std::multiset<int> s[N];
inline int fir(cs std::multiset<int> &s){return s.size()?*s.begin():0x3f3f3f3f;}

int fa[N],son[N][2],siz[N],lmn[N],rmn[N];
bool col[N];
inline void pushup(int u){
	int lc=son[u][0],rc=son[u][1];
	siz[u]=siz[lc]+siz[rc]+1;
	lmn[u]=min(lmn[lc],siz[lc]+min(col[u]?0:0x3f3f3f3f,min(fir(s[u]),lmn[rc]+1)));
	rmn[u]=min(rmn[rc],siz[rc]+min(col[u]?0:0x3f3f3f3f,min(fir(s[u]),rmn[lc]+1)));
}

inline bool which(int u){return son[fa[u]][1]==u;}
inline bool isroot(int u){return son[fa[u]][0]!=u&&son[fa[u]][1]!=u;}

inline void Rotate(int u){
	int Fa=fa[u],FA=fa[Fa];
	bool pos=which(u);
	if(!isroot(Fa))son[FA][which(Fa)]=u;
	son[Fa][pos]=son[u][!pos];
	if(son[u][!pos])fa[son[Fa][pos]]=Fa;
	son[u][!pos]=Fa;
	fa[u]=FA,fa[Fa]=u;
	pushup(Fa);
	pushup(u);
}

inline void Splay(int u){
	for(int re Fa=fa[u];!isroot(u);Rotate(u),Fa=fa[u])
	if(!isroot(Fa))Rotate(which(Fa)==which(u)?Fa:u);
}

inline void access(int u){
	for(int re ch=0;u;u=fa[ch=u]){
		Splay(u);
		s[u].insert(lmn[son[u][1]]+1);
		son[u][1]=ch;
		s[u].erase(s[u].find(lmn[ch]+1));
		pushup(u);
	}
}

inline void dfs(int u){
	for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])
	if(v!=fa[u]){
		fa[v]=u;
		s[u].insert(0x3f3f3f3f+1);
		dfs(v);
	}
	pushup(u);
}

signed main(){
	lmn[0]=rmn[0]=0x3f3f3f3f;
	n=getint();
	for(int re i=1;i<n;++i)addedge(getint(),getint());
	dfs(1);
	q=getint();
	while(q--)switch(getint()){
		case 1:{
			int u=getint();
			access(u);
			Splay(u);
			cout<<(rmn[u]>n?-1:rmn[u])<<"\n";
			break;
		}
		case 0:{
			int u=getint();
			access(u);
			Splay(u);
			col[u]^=1;
			pushup(u);
			break;
		}
	} 
	return 0;
}