[洛谷P1501] [国家集训队] Tree II

比之前那道LCT裸题难了一点儿。

题目传送门

这道题除了维护树的形态和连通性，还要维护点权了。

所以记录size、sum什么的，跟splay差不多。

lazy_tag、pushup、pushdown什么的，跟线段树差不多。

就是一道大杂烩嘛......

各种乱搞。

这里不再写cut(x,y)函数了，而是写了isolate(x,y)函数。

意为把从x到y的链单独提取到一个splay里面。

再加个删除父子关系不就是cut了嘛。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define M 51061
#define ll long long
#define id(x) (s[f[x]][1]==x)
using namespace std;

int n,q;
int s[100005][2],f[100005],sz[100005];
bool rev[100005],rt[100005];
ll v[100005],sum[100005],lza[100005],lzm[100005];

void reverse(int p)
{
    if(!p)return;
    swap(s[p][0],s[p][1]);
    rev[p]^=1;
}

void pushup(int p)
{
    sz[p]=sz[s[p][0]]+sz[s[p][1]]+1;
    sum[p]=(sum[s[p][0]]+sum[s[p][1]]+v[p])%M;
}

void mul(int p,ll c)
{
    sum[p]=sum[p]*c%M;
    v[p]=v[p]*c%M;
    lzm[p]=lzm[p]*c%M;
    lza[p]=lza[p]*c%M;
}

void plus(int p,ll c)
{
    sum[p]=(sum[p]+sz[p]*c%M)%M;
    v[p]=(v[p]+c)%M;
    lza[p]=(lza[p]+c)%M;
}

void pushdown(int p)
{
    if(lzm[p]!=1)mul(s[p][0],lzm[p]),mul(s[p][1],lzm[p]),lzm[p]=1;
    if(lza[p])plus(s[p][0],lza[p]),plus(s[p][1],lza[p]),lza[p]=0;
    if(rev[p])reverse(s[p][0]),reverse(s[p][1]),rev[p]=0;
}

void rotate(int p)
{
    int k=id(p);
    int fa=f[p];
    if(rt[fa])rt[p]=1,rt[fa]=0;
    else s[f[fa]][id(fa)]=p;
    s[fa][k]=s[p][!k];
    s[p][!k]=fa;
    f[p]=f[fa];
    f[fa]=p;
    f[s[fa][k]]=fa;
    pushup(fa);
    pushup(p);
}

void down(int p)
{
    if(!rt[p])down(f[p]);
    pushdown(p);
}

void splay(int p)
{
    down(p);
    while(!rt[p])
    {
        int fa=f[p];
        if(rt[fa])
        {
            rotate(p);
            return;
        }
        if(id(p)^id(fa))rotate(p);
        else rotate(fa);
        rotate(p);
    }
}

void access(int p)
{
    int son=0;
    while(p)
    {
        splay(p);
        rt[s[p][1]]=1,rt[son]=0;
        s[p][1]=son;
        pushup(p);
        son=p,p=f[p];
    }
}

void mtr(int p)
{
    access(p);
    splay(p);
    reverse(p);
}

void link(int x,int y)
{
    mtr(x);
    f[x]=y;
}

void isolate(int x,int y)
{
    mtr(x);
    access(y);
    splay(y);
}

int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)lzm[i]=sz[i]=v[i]=rt[i]=1;
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        link(x,y);
    }
    while(q--)
    {
        char op[5];
        scanf("%s",op+1);
        int pu,pv,nu,nv;
        ll c;
        if(op[1]=='+')
        {
            scanf("%d%d%lld",&pu,&pv,&c);
            isolate(pu,pv);
            plus(pv,c);
        }
        if(op[1]=='-')
        {
            scanf("%d%d%d%d",&pu,&pv,&nu,&nv);
            isolate(pu,pv);
            s[pv][0]=f[pu]=0;
            rt[pu]=1;
            link(nu,nv);
        }
        if(op[1]=='*')
        {
            scanf("%d%d%lld",&pu,&pv,&c);
            isolate(pu,pv);
            mul(pv,c);
        }
        if(op[1]=='/')
        {
            scanf("%d%d",&pu,&pv);
            isolate(pu,pv);
            printf("%lld\n",sum[pv]);
        }
    }
    return 0;
}