bzoj 3282 Link Cut Tree (LCT)

题目大意:维护一个森林,支持边的断,连,修改某个点的权值,求树链所有点点权的异或和

洛谷P3690传送门

搞了一个下午终于明白了LCT的原理

#include <cstdio>
#include <algorithm>
#include <cstring>
#define root d[0].ch[1]
#define il inline
#define nu 7777
#define inf 500000
#define N 300100
using namespace std;

int n,m,tp;
int stk[N];
char str[10];
struct Link_Cut_Tree{
    int fa[N],ch[N][2],sum[N],val[N],rev[N];
    il int idf(int x){return ch[fa[x]][0]==x?0:1;}
    il int isroot(int x){return (ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x)?1:0;}
    il void con(int x,int ff,int p){fa[x]=ff;ch[ff][p]=x;}
    il void pushup(int x){sum[x]=sum[ch[x][0]]^sum[ch[x][1]]^val[x];}
    il void revers(int x){swap(ch[x][0],ch[x][1]),rev[x]^=1;}
    il void pushdown(int x){
        if(rev[x]){ 
            if(ch[x][0]) revers(ch[x][0]);
            if(ch[x][1]) revers(ch[x][1]);
            rev[x]=0;}}
    il void rot(int x){
        int y=fa[x];int ff=fa[y];int px=idf(x);int py=idf(y);
        if(!isroot(y)) ch[ff][py]=x;
        fa[x]=ff,con(ch[x][px^1],y,px),con(y,x,px^1);
        pushup(y),pushup(x);}
    void splay(int x){
        int y=x;stk[++tp]=x;
        while(!isroot(y)){stk[++tp]=fa[y];y=fa[y];}
        while(tp){pushdown(stk[tp--]);}
        while(!isroot(x)){
            y=fa[x];
            if(isroot(y)) rot(x);
            else if(idf(y)==idf(x)) rot(y),rot(x);
            else rot(x),rot(x);
        }}
    il void access(int x){for(int y=0;x;y=x,x=fa[x])splay(x),ch[x][1]=y,pushup(x);}
    il void mkroot(int x){access(x),splay(x),revers(x);}
    il int findrt(int x){
        access(x),splay(x);
        while(ch[x][0])pushdown(x),x=ch[x][0];
        return x;}
    il void split(int x,int y){mkroot(x),access(y),splay(y);} 
    il void link(int x,int y){mkroot(x);if(findrt(y)!=x)fa[x]=y;}
    il void cut(int x,int y){
        mkroot(x);
        if(findrt(y)==x&&fa[x]==y&&!ch[x][1])
            fa[x]=ch[y][0]=0,pushup(y);}
}lct;

int gc()
{
    int rett=0,fh=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')fh=-1;c=getchar();}
    while(c>='0'&&c<='9'){rett=(rett<<3)+(rett<<1)+c-'0';c=getchar();}
    return rett*fh;
}

int main()
{
    n=gc(),m=gc();
    for(int i=1;i<=n;i++) lct.val[i]=gc();
    int fl,x,y;
    for(int i=1;i<=m;i++)
    {
        fl=gc(),x=gc(),y=gc();
        if(fl==0) lct.split(x,y),printf("%d\n",lct.sum[y]);
        if(fl==1) lct.link(x,y);
        if(fl==2) lct.cut(x,y);
        if(fl==3) lct.splay(x),lct.val[x]=y,lct.pushup(x);
    }
    return 0;
}



猜你喜欢

转载自blog.csdn.net/guapisolo/article/details/82695011