【洛谷】3690 【模板】Link Cut Tree （动态树）

题目传送门

一道迟到的模板题。话说在思路清晰的时候敲代码真的爽啊。

修改操作可以先访问该节点，然后把它旋转到根，这样只用修改这一个点的权值就好了。

询问操作可以把其中一个节点搞成根，然后两点间的权值就变成了另一个节点到根的权值了。

这两个想法挺妙。

附上AC代码：

#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;

const int N=3e5+10;
int n,m,w[N],sum[N],x,y,o,mk[N],ch[N][2],f[N];

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

#define isrt(x) ((x)!=ch[f[x]][0]&&(x)!=ch[f[x]][1])
inline void updata(int x){sum[x]=sum[ch[x][0]]^sum[ch[x][1]]^w[x];}
inline void rotate(int x){
    int y=f[x],op=(x==ch[y][1]);
    ch[y][op]=ch[x][op^1];
    if (ch[x][op^1]) f[ch[x][op^1]]=y;
    f[x]=f[y];
    if (!isrt(y)) ch[f[y]][ch[f[y]][1]==y]=x;
    f[y]=x,ch[x][op^1]=y;
    return updata(y),updata(x);
}
inline void push(int x){if (mk[x]) mk[x]=0,mk[ch[x][0]]^=1,mk[ch[x][1]]^=1,swap(ch[x][0],ch[x][1]);}
inline void splay(int x){
    static int top,sk[N];sk[top=1]=x;
    for (int i=x; !isrt(i); i=f[i]) sk[++top]=f[i];
    while (top) push(sk[top--]);
    for (int fa=f[x]; !isrt(x); rotate(x),fa=f[x])
        if (!isrt(fa)) rotate((x==ch[fa][0])==(fa==ch[f[fa]][0])?fa:x);
}
inline void access(int x){for (int lst=0; x; lst=x,x=f[x]) splay(x),ch[x][1]=lst,updata(x);}
inline void root(int x){access(x),splay(x),mk[x]^=1;}
inline void add(int x,int y){root(x),f[x]=y;}
inline void del(int x,int y){root(x),access(y),splay(y),f[x]=ch[y][0]=0;}
inline int gf(int x){access(x),splay(x); while (ch[x][0]) x=ch[x][0]; return x;}

int main(void){
    read(n),read(m);
    for (int i=1; i<=n; ++i) read(w[i]),sum[i]=w[i];
    while (m--){
        read(o),read(x),read(y);
        switch (o){
            case 0: root(x),access(y),splay(y),printf("%d\n",sum[y]); break;
            case 1: if (gf(x)!=gf(y)) add(x,y); break;
            case 2: if (gf(x)==gf(y)) del(x,y); break;
            case 3: access(x),splay(x),w[x]=y,updata(x); break;
        }
    }
    return 0;
}