BZOJ2631: tree

给棵树，支持：连边删边，链加，链乘，链求和。

LCT？？

#include<string.h>
#include<stdlib.h>
#include<stdio.h>
#include<math.h>
//#include<assert.h>
#include<algorithm> 
//#include<iostream>
//#include<bitset>
using namespace std;

int n,m;
#define maxn 100011
const int mod=51061;
struct LCT
{
    struct Node
    {
        int fa,son[2];
        bool rev; int add,mul;
        int size,v,sum;
    }a[maxn];
    void makeatree(int id)
    {
        a[id].fa=a[id].son[0]=a[id].son[1]=a[id].rev=0;
        a[id].add=0; a[id].v=a[id].sum=1; a[id].mul=a[id].size=1;
    }
    void up(int x)
    {
        if (!x) return;
        int &p=a[x].son[0],&q=a[x].son[1];
        a[x].sum=(a[p].sum+a[q].sum+a[x].v)%mod;
        a[x].size=a[p].size+a[q].size+1;
    }
    void revsingle(int x)
    {
        if (!x) return;
        a[x].rev^=1;
        int &p=a[x].son[0],&q=a[x].son[1];
        p^=q; q^=p; p^=q;
    }
    void addsingle(int x,int v)
    {
        if (!x) return;
        a[x].v+=v; a[x].v-=a[x].v>=mod?mod:0;
        a[x].add+=v; a[x].add-=a[x].add>=mod?mod:0;
        a[x].sum=(a[x].sum+1ll*a[x].size*v)%mod;
    }
    void mulsingle(int x,int v)
    {
        if (!x) return;
        a[x].v=1ll*a[x].v*v%mod;
        a[x].mul=1ll*a[x].mul*v%mod;
        a[x].add=1ll*a[x].add*v%mod;
        a[x].sum=1ll*a[x].sum*v%mod;
    }
    bool isroot(int x)
    {
        if (!a[x].fa) return 1;
        return (a[a[x].fa].son[0]!=x && a[a[x].fa].son[1]!=x);
    }
    void down(int x)
    {
        int &p=a[x].son[0],&q=a[x].son[1];
        if (a[x].rev) {revsingle(p); revsingle(q); a[x].rev=0;}
        if (a[x].mul!=1) {mulsingle(p,a[x].mul); mulsingle(q,a[x].mul); a[x].mul=1;}
        if (a[x].add) {addsingle(p,a[x].add); addsingle(q,a[x].add); a[x].add=0;}
    }
    int sta[maxn];
    void download(int x)
    {
        int top=0; for (;!isroot(x);x=a[x].fa) sta[++top]=x; sta[++top]=x;
        for (;top;top--) down(sta[top]);
    }
    void rotate(int x)
    {
        const int y=a[x].fa,z=a[y].fa;
        bool w=(x==a[y].son[0]);
        a[x].fa=z;
        if (!isroot(y)) a[z].son[y==a[z].son[1]]=x;
        a[y].son[w^1]=a[x].son[w];
        if (a[x].son[w]) a[a[x].son[w]].fa=y;
        a[x].son[w]=y;
        a[y].fa=x;
        up(y); up(z);
    }
    void splay(int x)
    {
        if (!x) return;
        download(x);
        while (!isroot(x))
        {
            const int y=a[x].fa,z=a[y].fa;
            if (!isroot(y))
            {
                if ((y==a[z].son[0])^(x==a[y].son[0])) rotate(x);
                else rotate(y);
            }
            rotate(x);
        }
        up(x);
    }
    void access(int x)
    {
        int y=0,tmp=x;
        while (x) {splay(x); a[x].son[1]=y; up(x); y=x; x=a[x].fa;}
        splay(tmp);
    }
    void resetroot(int x)
    {
        access(x);
        revsingle(x);
    }
    void link(int x,int y)
    {
        resetroot(x);
        a[x].fa=y;
    }
    void cut(int x,int y)
    {
        resetroot(x);
        access(y);
        a[y].son[0]=a[x].fa=0;
        up(y);
    }
    void Add(int x,int y,int v)
    {
        resetroot(x);
        access(y);
        addsingle(y,v);
    }
    void Mul(int x,int y,int v)
    {
        resetroot(x);
        access(y);
        mulsingle(y,v);
    }
    int query(int x,int y)
    {
        resetroot(x);
        access(y);
        return a[y].sum;
    }
}t;

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) t.makeatree(i);
    for (int i=1,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        t.link(x,y);
    }
    char c;int x,y,w,z;
    while (m--)
    {
        while ((c=getchar())!='+' && c!='-' && c!='*' && c!='/');
        if (c=='+')
        {
            scanf("%d%d%d",&x,&y,&z);
            t.Add(x,y,z);
        }
        else if (c=='*')
        {
            scanf("%d%d%d",&x,&y,&z);
            t.Mul(x,y,z);
        }
        else if (c=='-')
        {
            scanf("%d%d%d%d",&x,&y,&z,&w);
            t.cut(x,y); t.link(z,w);
        }
        else if (c=='/')
        {
            scanf("%d%d",&x,&y);
            printf("%d
",t.query(x,y));
        }
    }
    return 0;
}