LG3690 【【模板】Link Cut Tree (动态树)】

题目

终于去写\(LCT\)

这个大爷讲的挺好的

板子

#include<iostream>
#include<cstdio>
#define maxn 10005
#define re register
inline int read()
{
    char c=getchar();int x=0;while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
char opt[15];
int n,m;
int rev[maxn],fa[maxn],ch[maxn][2],st[maxn];
inline int nroot(int x) {return ch[fa[x]][0]==x||ch[fa[x]][1]==x;}
inline void pushdown(int x) {
    if(!rev[x]) return;
    rev[x]=0,rev[ch[x][0]]^=1,rev[ch[x][1]]^=1;
    std::swap(ch[ch[x][0]][0],ch[ch[x][0]][1]);
    std::swap(ch[ch[x][1]][0],ch[ch[x][1]][1]);
}
inline void rotate(int x) {
    int y=fa[x],z=fa[y],k=ch[y][1]==x,w=ch[x][k^1];
    if(nroot(y)) ch[z][ch[z][1]==y]=x;
    ch[x][k^1]=y,ch[y][k]=w;
    fa[w]=y;fa[y]=x,fa[x]=z;
}
inline void splay(int x) {
    int y=x,top=0;
    st[++top]=x;
    while(nroot(y)) st[++top]=fa[y],y=fa[y];
    while(top) pushdown(st[top--]);
    while(nroot(x)) {
        int y=fa[x];
        if(nroot(y)) rotate((ch[fa[y]][1]==y)^(ch[y][1]==x)?x:y);
        rotate(x);
    }
}
inline void access(int x) {
    for(re int y=0;x;y=x,x=fa[x])
        splay(x),ch[x][1]=y;
}
inline void makeroot(int x) {
    access(x);splay(x);rev[x]^=1;std::swap(ch[x][0],ch[x][1]);
}
inline int findroot(int x) {
    access(x),splay(x);
    while(ch[x][0]) pushdown(x),x=ch[x][0];
    return x;
}
inline void link(int x,int y) {
    makeroot(x);
    if(findroot(y)!=x) fa[x]=y;
}
inline void split(int x,int y) {
    makeroot(x);access(y);splay(y);
}
inline void cut(int x,int y) {
    split(x,y);fa[x]=ch[y][0]=0;
}
int main()
{
    n=read(),m=read();
    int x,y;
    while(m--) {
        scanf("%s",opt);x=read(),y=read();
        if(opt[0]=='D') cut(x,y);
        if(opt[0]=='C') link(x,y);
        if(opt[0]=='Q') puts((findroot(x)==findroot(y))?"Yes":"No");
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/asuldb/p/10366716.html
今日推荐