题目
维护一个森林,支持连边操作和查询操作。
解题思路
直接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;
}