bzoj2049 [Sdoi2008]Cave 洞穴勘测 link cut tree入门

link cut tree入门题

首先说明本人只会写自底向上的数组版(都说了不写指针、不写自顶向下QAQ……)

突然发现link cut tree不难写。。。

说一下各个函数作用：

bool isroot(int x):判断x是否为所在重链(splay)的根

void down(int x):下放各种标记

void rotate(int x):在x所在重链(splay)中将x旋转到fa[x]的位置上

void splay(int x):在x坐在重链(splay)中将x旋转到根

void access(int x):把x到x所在树的根变为一条重链，根为链顶，x为链底

void reverse(int x):把x变为所在树的根(先access再打上reverse标记)

void link(int x,int y)连接x,y两点

void cut(int x,int y)断开x,y两点

int find(int x)找到x所在树的根

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define rre(i,r,l) for(int i=(r);i>=(l);i--)
#define re(i,l,r) for(int i=(l);i<=(r);i++)
#define Clear(a,b) memset(a,b,sizeof(a))
#define inout(x) printf("%d",(x))
#define douin(x) scanf("%lf",&x)
#define strin(x) scanf("%s",(x))
#define LLin(x) scanf("%lld",&x)
#define op operator
#define CSC main
typedef unsigned long long ULL;
typedef const int cint;
typedef long long LL;
using namespace std;
void inin(int &ret)
{
    ret=0;int f=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
    while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
    ret=f?-ret:ret;
}
int fa[10010],ch[10010][2],rev[10010];
bool isroot(int x){return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;}
void down(int x)
{
    if(!rev[x])return ;
    swap(ch[x][0],ch[x][1]);
    rev[ch[x][0]]^=1;
    rev[ch[x][1]]^=1;
    rev[x]=0;
}
void rotate(int x)
{
    int y=fa[x],z=fa[y];
    if(!isroot(y))
        if(ch[z][0]==y)ch[z][0]=x;
        else ch[z][1]=x;else ;
    fa[x]=z,fa[y]=x;
    int d=ch[y][1]==x;
    fa[ch[x][d^1]]=y;
    ch[y][d]=ch[x][d^1];
    ch[x][d^1]=y;
}
int sta[10010],top;
void splay(int x)
{
    top=0;sta[++top]=x;
    for(int i=x;!isroot(i);i=fa[i])sta[++top]=fa[i];
    while(top)down(sta[top--]);
    while(!isroot(x))
    {
        int y=fa[x],z=fa[y];
        if(!isroot(y))
            if((ch[y][1]==x)^(ch[z][1]==y))rotate(x);
            else rotate(y);else 
        rotate(x);
    }
}
void access(int x)
{
    int temp=0;
    while(x)
    {
        splay(x);
        ch[x][1]=temp;
        temp=x,x=fa[x];
    }
}
void reverse(int x)
{
    access(x);splay(x),rev[x]^=1;
}
void link(int x,int y)
{
    reverse(x);fa[x]=y,splay(x);
}
void cut(int x,int y)
{
    reverse(x);access(y);splay(y);ch[y][0]=fa[x]=0;
}
int find(int x)
{
    access(x),splay(x);
    int y=x;
    while(ch[y][0])y=ch[y][0];
    return y;
}
int n,m;
int main()
{
    char s[11];
    inin(n),inin(m);
    re(i,1,m)
    {
        strin(s);
        int x,y;inin(x),inin(y);
        if(s[0]=='C')link(x,y);
        else if(s[0]=='D')cut(x,y);
        else 
        {
            if(find(x)==find(y))printf("Yes
");
            else printf("No
");
        }
    }
     return 0;
}