JZOJ 5883 到不了

题目

维护一个森林，支持连边操作和查询操作。

解题思路

直接LCT做即可。
LCT求LCA(u,v)？
有2种情况。如果v不是u的祖先，那么：
access(u)将u到根的路径标示出来，在access(v)的时候，目前splay树的根为x，则最后一个fa[x]就是LCA(u,v)。
如果v是u的祖先，那么：
判断一下这种情况，LCA(u,v)=v。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 100010
#define P(a) putchar(a)
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
struct note{
	int rev,f;
	int c[2];
}tr[N];
struct notd{
	int to,next;
}edge[N<<1];
int tot,head[N];
int i,j,k,n,m,q,cs,op,CNT;
int u,v,ans;
int rt[N],fa[N],bel[N];
int read(){
	int rs=0,fh=1;char ch;
	while((ch<'0'||ch>'9')&&(ch^'-'))ch=getchar();
	if(ch=='-')fh=-1,ch=getchar();
	while(ch>='0'&&ch<='9')rs=(rs<<3)+(rs<<1)+(ch^'0'),ch=getchar();
	return rs*fh;
}
void write(int x){
	if(x>9)write(x/10);
	P(x%10+'0'); 
}
int Max(int x,int y){return x>y?x:y;}
void lb(int x,int y){
	edge[++tot].to=y;
	edge[tot].next=head[x];
	head[x]=tot;
}
void Swap(int &x,int &y){x^=y;y^=x;x^=y;}
void pdw(int x){
	if(tr[x].rev){
	    int ls=tr[x].c[0],rs=tr[x].c[1];
	    if(ls){
	    	tr[ls].rev^=1;
	    	Swap(tr[ls].c[0],tr[ls].c[1]);
		}
		if(rs){
	    	tr[rs].rev^=1;
	    	Swap(tr[rs].c[0],tr[rs].c[1]);
		}
		tr[x].rev=0;
	}
}
bool isroot(int x){return (tr[tr[x].f].c[0]^x)&&(tr[tr[x].f].c[1]^x);}
bool dir(int x){return tr[tr[x].f].c[1]==x;}
int get(int x){return bel[x]==x?x:bel[x]=get(bel[x]);} 
void rotate(int x){
	int y=tr[x].f,z=dir(x);
	if(!isroot(y))tr[tr[y].f].c[dir(y)]=x;
	tr[y].c[z]=tr[x].c[z^1];
	tr[x].f=tr[y].f;
	tr[y].f=x;
	tr[tr[x].c[z^1]].f=y;
	tr[x].c[z^1]=y;
}
void update(int x){
	if(!isroot(x))update(tr[x].f);
	pdw(x);
}
void splay(int x){
	int y;
	update(x);
	while(!isroot(x)){
		y=tr[x].f;
		if(!isroot(y)){
			if(dir(x)^dir(y))rotate(x);else rotate(y);
		}
		rotate(x);
	}
}
void access(int x){
	int y=0;
	while(x){
		splay(x);
		pdw(x);
		tr[x].c[1]=y;
		y=x;
		x=tr[x].f;
	}
}
void access1(int x){
	int y=0,z=-1;
	while(x){
		splay(x);
		pdw(x);
		tr[x].c[1]=y;
		y=x;
		if(tr[x].f)z=tr[x].f;
		else{
			if(z==-1)ans=v;
		}
		x=tr[x].f;
	}
	if(z!=-1)ans=z;
}
void mroot(int x){
	access(x);
	splay(x);
	Swap(tr[x].c[0],tr[x].c[1]);
	tr[x].rev=1;
}
void link(int u,int v){
	mroot(u);
	mroot(v);
	tr[v].f=u;
}
void dg(int x){
	int i;
	bel[x]=rt[CNT];
	for(i=head[x];i;i=edge[i].next)
	    if(fa[x]^edge[i].to){
	    	fa[edge[i].to]=x;
	    	link(x,edge[i].to);
	    	dg(edge[i].to);
		}
}
int main(){
    n=read();m=read();
    fo(i,1,m)rt[i]=read();
    q=read();
    fo(i,1,n-m){
    	u=read();v=read();
    	lb(u,v);lb(v,u);
	}
	fo(i,1,m){
		CNT=i;
		dg(rt[i]);
		mroot(rt[i]);
	}
	int gx,gy;
    q=read();
    fo(cs,1,q){
    	op=read();u=read();v=read();
    	if(op==1){
    		gx=get(u);
    		gy=get(v);
    		if(gx==gy)continue;
    		link(u,v);
    		bel[gy]=gx;
    		gx=get(u);
    		gy=get(v);
    		mroot(bel[u]);
		}else{
			gx=get(u);
    		gy=get(v);
    		if(bel[gx]!=bel[gy]){
    			printf("orzorz\n");
    			continue;
			}
			access(u);
			access1(v);
			write(ans),P('\n');
		}
	}
	return 0;
}