動態樹LCT 模板

題目描述：

輸入：
第一行兩個整數n和m；
接下來一行中n個整數表示初始點權；
接下來m行每行一個操作如上表所示。

輸出：
對於每一個連接操作，若p和q不連通，輸出YES，並添加這條邊；否則輸出NO；
對於每一個刪除操作，若p和q間有邊，輸出YES，並刪除這條邊，否則輸出NO；
對於每一個查詢最大及查詢和，若p和q連通，輸出一行包含一個整數爲對應的答案；否則輸出一個整數-1。

樣例輸入：
5 10
1 3 5 4 6
LINK 1 2
LINK 1 3
LINK 2 3
MAX 2 3
UPDATE 1 6
MAX 2 3
LINK 3 5
SUM 3 5
CUT 1 3
SUM 1 5

樣例輸出：
YES
YES
NO
5
6
YES
11
YES
-1

代碼如下：

#include <cstdio>
#include <algorithm>
#define N 50000
using namespace std;
inline int Max(int x,int y) { return x>y?x:y; }
int n,m,x,y;
char s[10];
struct splay{
    splay *ch[2],*fa;
    int val,sum,max_val;
    bool mark;
    splay(int x);
    void maintain();
    int dir();
    void Reverse();
    void push_down();
    void push_up();
}*null=new splay(0),*root[N];
splay :: splay(int x)
{
    val=sum=max_val=x;
    ch[0]=ch[1]=fa=null;
    mark=false;
}
void splay :: maintain()
{
    sum=ch[0]->sum+ch[1]->sum+val;
    max_val=Max(val,Max(ch[0]->max_val,ch[1]->max_val));
}
int splay :: dir()
{
    return fa->ch[0]==this?0:(fa->ch[1]==this?1:-1);
}
void splay :: Reverse()
{
    if(this==null) return ;
    mark=!mark;
    swap(ch[0],ch[1]);
    return ;
}
void splay :: push_down()
{
    if(this==null) return;
    if(mark)
    {
        ch[0]->Reverse();
        ch[1]->Reverse();
        mark=false;
    }
    return;
}
void splay :: push_up()
{
    if(~dir()) fa->push_up();
    push_down();
}
void turn(splay *c,int d)
{
    splay *y=c->ch[d^1];
    c->ch[d^1]=y->ch[d];
    if(y->ch[d]!=null) y->ch[d]->fa=c;
    y->ch[d]=c;
    y->fa=c->fa;
    int k;
    if(~(k=c->dir())) c->fa->ch[k]=y;
    c->fa=y;
    c->maintain();
    y->maintain();
}
void splaying(splay *c)
{
    c->push_up();
    int d;
    while(~(d=c->dir()))
    {
        if(d==c->fa->dir()) turn(c->fa->fa,d^1);
        turn(c->fa,d^1);
    }
    return;
}
void Access(splay *c)
{
    splay *tmp=null;
    while(c!=null)
    {
        splaying(c);
        c->ch[1]=tmp; c->maintain();
        tmp=c;
        c=c->fa;
    }
    return;
}
void Move_to_root(splay *c)
{
    Access(c),splaying(c);
    c->Reverse();
    return;
}
void Link(splay *x,splay *y)
{
    Move_to_root(x);
    Access(y),splaying(y);
    if(x->fa!=null) { printf("NO\n"); return; }
    printf("YES\n");
    x->fa=y;
    return;
}
void Cut(splay *x,splay *y)
{
    Move_to_root(x);
    Access(y),splaying(y);
    if(y->ch[0]!=x || x->ch[1]!=null) { printf("NO\n"); return; }
    printf("YES\n");
    y->ch[0]=null; y->maintain();
    x->fa=null;
    return;
}
void query_Max(splay *x,splay *y)
{
    Move_to_root(x);
    Access(y); splaying(y);
    if(x->fa==null) printf("-1\n");
    else printf("%d\n",y->max_val);
    return;
}
void query_Sum(splay *x,splay *y)
{
    Move_to_root(x);
    Access(y); splaying(y);
    if(x->fa==null) printf("-1\n");
    else printf("%d\n",y->sum);
    return;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        root[i]=new splay(x);
    }
    while(m--)
    {
        scanf("%s%d%d",s,&x,&y);
        switch(s[0])
        {
            case 'U':
                splaying(root[x]);
                root[x]->val=y;
                root[x]->maintain();
                break;
            case 'L':
                Link(root[x],root[y]);
                break;
            case 'C':
                Cut(root[x],root[y]);
                break;
            case 'M':
                query_Max(root[x],root[y]);
                break;
            case 'S':
                query_Sum(root[x],root[y]);
                break;

        }
    }
    return 0;
}