LCT
简单的LCT应用。
每个询问其实就是询问他们的祖先是否相等。只需要access(x),splay(x),再不断往左走到最底就是x的祖先了。
丑陋的代码:
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 10005
#define il inline
#define iv il void
using namespace std;
struct node{
int fa,to[2],f;
}t[N];
int n,m,stk[N],tp;
char s[10];
il int _read(){
int x=0; char ch=getchar();
while (!isdigit(ch)) ch=getchar();
while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
return x;
}
iv pshd(int x){
if (t[x].f){
int &l=t[x].to[0],&r=t[x].to[1];
t[l].f^=1,t[r].f^=1,t[x].f=0,swap(l,r);
}
}
il bool rt(int x){ return t[t[x].fa].to[0]!=x&&t[t[x].fa].to[1]!=x; }
iv rtt(int x){
int y=t[x].fa,z=t[y].fa,l=(t[y].to[1]!=x);
if (!rt(y)) t[z].to[t[z].to[1]==y]=x;
t[x].fa=z,t[y].fa=x,t[t[x].to[l]].fa=y;
t[y].to[l^1]=t[x].to[l],t[x].to[l]=y;
}
iv splay(int x){
stk[tp=1]=x;
for (int i=x;!rt(i);i=t[i].fa) stk[++tp]=t[i].fa;
for (int i=tp;i;i--) pshd(stk[i]);
while (!rt(x)){
int y=t[x].fa,z=t[y].fa;
if (!rt(y)) if (t[z].to[0]==y^t[y].to[0]==x) rtt(x); else rtt(y);
rtt(x);
}
}
iv ccss(int x){ for (int i=0;x;i=x,x=t[x].fa) splay(x),t[x].to[1]=i; }
iv mkrt(int x){ ccss(x),splay(x),t[x].f^=1; }
iv lnk(int x,int y){ mkrt(x),t[x].fa=y; }
iv sprt(int x,int y){ mkrt(x),ccss(y),splay(y); }
iv cut(int x,int y){ sprt(x,y),t[y].to[0]=t[x].fa=0; }
il int find(int x){ ccss(x),splay(x); while (t[x].to[0]) x=t[x].to[0]; return x; }
int main(){
n=_read(),m=_read();
while (m--){
scanf("%s",s); int x=_read(),y=_read();
switch(s[0]){
case 'Q':
if (find(x)==find(y)) printf("Yes\n");
else printf("No\n"); break;
case 'C': lnk(x,y); break;
case 'D': cut(x,y); break;
}
}
return 0;
}