【bzoj2049】洞穴勘测 lct

    好久不更了。但是学新算法,就算做作业时间再紧张,也要记一下。

    初学lct。几个要点提一下

      1.同一个联通块是同一棵。不是一个splay或者一条链;

      2.根节点有很多。刚开始打还想着用变量rt我是傻了吗;

      3.那个access还算好理解。关键是代码里面那个rever是把x变为x所在树(不是splay)的根节点,理解了好久;

    差不多就这么多了吧。

代码是抄hzwer黄学长的。orz~:

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 30025
using namespace std;
 
int n,m,c[N][2],fa[N],h[N];
bool rev[N];
inline int read(){
    int x=0,f=1; char ch=getchar();
    while (ch==' ' || ch=='\r' || ch=='\n') ch=getchar();
    if (ch=='+'){ f=1; ch=getchar(); }
    if (ch=='-'){ f=-1; ch=getchar(); }
    while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
    return x*f;
}
bool isrt(int x){
    return c[fa[x]][0]!=x && c[fa[x]][1]!=x;
}
void pushdown(int x){
    int l=c[x][0],r=c[x][1];
    if (rev[x]){
        rev[x]=0; rev[l]^=1; rev[r]^=1;
        swap(c[x][0],c[x][1]);
    }
}
void rotate(int x){
    int y=fa[x],z=fa[y],l,r;
    l=(c[y][0]==x)?0:1; r=l^1;
    if (!isrt(y)){
        if (c[z][0]==y) c[z][0]=x; else c[z][1]=x;
    }
    fa[x]=z; fa[y]=x; fa[c[x][r]]=y;
    c[y][l]=c[x][r]; c[x][r]=y;
}
void splay(int x){
    int top=1,p,i,y,z; h[1]=x;
    for (p=x; !isrt(p); p=fa[p]) h[++top]=fa[p];
    for (i=top; i; i--) pushdown(h[i]);
    while (!isrt(x)){
        y=fa[x]; z=fa[y];
        if (!isrt(y)){
            if ((c[y][0]==x)^(c[z][0]==y))
                rotate(x); else rotate(y);
        }
        rotate(x);
    }
}
void access(int x){
    int y=0;
    while (x){
        splay(x); c[x][1]=y;
        y=x; x=fa[x];
    }
}
void rever(int x){
    access(x); splay(x); rev[x]^=1;
}
void link(int x,int y){
    rever(x); fa[x]=y; splay(x);
}
void cut(int x,int y){
    rever(x); access(y); splay(y); fa[x]=c[y][0]=0;
}
int find(int x){
    access(x); splay(x);
    int y=x;
    while (c[y][0]) y=c[y][0];
    return y;
}
int main(){
    n=read(); m=read();
    int i,x,y; char ch[10];
    for (i=1; i<=m; i++){
        scanf("%s",ch);
        x=read(); y=read();
        if (ch[0]=='C') link(x,y);
        else if (ch[0]=='D') cut(x,y);
        else puts((find(x)==find(y))?"Yes":"No");
    }
    return 0;
}


2015.8.2

by lych

你可能感兴趣的:(splay,LCT)