题目
Dynamic Tree Connectivity
分析
动态树(LCT)初探
题目
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
int read(){
int x=0;bool f=0;char c=getchar();
while(c<'0'||c>'9') f|=c=='-',c=getchar();
while(c>='0'&&c<='9') x=x*10+(c^48),c=getchar();
return f?-x:x;
}
#define IL inline
#define LL long long
#define MAXN 100000
#define INF 0x7fffffff
struct Node{
bool rev;
Node *fa,*ch[2];
}*NIL,Sp[MAXN+5];
IL bool Dir(Node *u){
return u->fa->ch[1]==u;
}
IL bool isRoot(Node *u){
return u->fa==NIL||(u->fa->ch[0]!=u&&u->fa->ch[1]!=u);
}
IL void SetChild(Node *u,Node *v,bool d){
u->ch[d]=v;
if(v!=NIL)
v->fa=u;
}
IL void PushDown(Node *u){
if(u->rev){
u->rev=0;
swap(u->ch[0],u->ch[1]);
u->ch[0]->rev^=1;
u->ch[1]->rev^=1;
}
}
IL void Rotate(Node *u){
bool d=Dir(u);
Node *y=u->fa;
if(isRoot(y))
u->fa=y->fa;
else
SetChild(y->fa,u,Dir(y));
SetChild(y,u->ch[!d],d);
SetChild(u,y,!d);
}
void Go(Node *u){
if(!isRoot(u))
Go(u->fa);
PushDown(u);
}
void Splay(Node *u){
Go(u);
while(!isRoot(u)){
Node *y=u->fa;
if(isRoot(y))
Rotate(u);
else{
if(Dir(u)==Dir(y))
Rotate(y);
else
Rotate(u);
Rotate(u);
}
}
}
Node *Access(Node *u){
Node *r=NIL;
while(u!=NIL){
Splay(u);
SetChild(u,r,1);
r=u,u=u->fa;
}
return r;
}
Node *GetRoot(Node *u){
Access(u);
Splay(u);
Node *r=u;
while(r->ch[0]!=NIL)
r=r->ch[0];
Splay(r);
return r;
}
IL void MakeRoot(Node *u){
Access(u);
Splay(u);
u->rev^=1;
}
IL void Link(Node *u,Node *v){
MakeRoot(u);
u->fa=v;
}
IL void Cut(Node *u,Node *v){
MakeRoot(u);
Access(v);
Splay(v);
u->fa=NIL;
v->ch[0]=NIL;
}
int main(){
int N=read(),M=read();
NIL=&Sp[0];
NIL->fa=NIL->ch[0]=NIL->ch[1]=NIL;
for(int i=1;i<=N;i++)
Sp[i]=*NIL;
while(M--){
char opt[10];
scanf("%s",opt);
Node *u=&Sp[read()],*v=&Sp[read()];
if(opt[0]=='a')
Link(u,v);
else if(opt[0]=='r')
Cut(u,v);
else
puts(GetRoot(u)==GetRoot(v)?"YES":"NO");
}
}