【bzoj2631】tree

Description

　一棵n个点的树，每个点的初始权值为1。对于这棵树有q个操作，每个操作为以下四种操作之一：
+ u v c：将u到v的路径上的点的权值都加上自然数c；
– u1 v1 u2 v2：将树中原有的边(u1,v1)删除，加入一条新边(u2,v2)，保证操作完之后仍然是一棵树；
* u v c：将u到v的路径上的点的权值都乘上自然数c；
/ u v：询问u到v的路径上的点的权值和，求出答案对于51061的余数。

Input

　　第一行两个整数n，q
接下来n-1行每行两个正整数u，v，描述这棵树
接下来q行，每行描述一个操作
Output

　　对于每个/对应的答案输出一行
Sample Input

3 2
1 2
2 3
* 1 3 4
/ 1 1
Sample Output

4
HINT

数据规模和约定
10%的数据保证，1<=n，q<=2000
另外15%的数据保证，1<=n，q<=5*10^4，没有-操作，并且初始树为一条链
另外35%的数据保证，1<=n，q<=5*10^4，没有-操作
100%的数据保证，1<=n，q<=10^5，0<=c<=10^4

题解
lct

代码

#include<bits/stdc++.h>
#define ll long long
#define inf 1000000007
#define mod 51061
#define lint unsigned int
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,top,cnt;
int c[100005][2],fa[100005];
int size[100005],q[100005];
bool rev[100005];
lint sum[100005],val[100005],at[100005],mt[100005];
inline bool isroot(int x){return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;}
inline void update(int x)
{
    int l=c[x][0],r=c[x][1];
    size[x]=(size[l]+size[r]+1)%mod;
    sum[x]=(sum[l]+sum[r]+val[x])%mod;
}
inline void cal(int x,int m,int a)
{
    if (!x) return;
    val[x]=(val[x]*m+a)%mod;
    sum[x]=(sum[x]*m+a*size[x])%mod;
    mt[x]=(mt[x]*m)%mod;
    at[x]=(at[x]*m+a)%mod;
}
inline void pushdown(int x)
{
    int l=c[x][0],r=c[x][1];
    if (rev[x])
    {
        rev[x]^=1;rev[l]^=1;rev[r]^=1;
        swap(c[x][0],c[x][1]);
    }
    int m=mt[x],a=at[x];
    mt[x]=1;at[x]=0;
    if (m!=1||a!=0)
    {
        cal(l,m,a);cal(r,m,a);
    }
}
inline void rotate(int x)
{
    int l,r,y=fa[x],z=fa[y];
    if (c[y][0]==x) l=0;else l=1;r=l^1;
    if (!isroot(y))
    {
        if (c[z][0]==y) c[z][0]=x;else c[z][1]=x;
    }
    fa[x]=z;fa[y]=x;
    c[y][l]=c[x][r];fa[c[x][r]]=y;
    c[x][r]=y;
    update(y);update(x);
}
void splay(int x)
{
    int top=0;
    q[++top]=x;
    for (int i=x;!isroot(i);i=fa[i])
        q[++top]=fa[i];
    while (top) pushdown(q[top--]);
    while (!isroot(x))
    {
        int y=fa[x],z=fa[y];
        if (!isroot(y))
        {
            if (c[y][0]==x^c[z][0]==y) rotate(x);else rotate(y);
        }
        rotate(x);
    }
}
void access(int x)
{
    int t=0;
    while (x)
    {
        splay(x);
        c[x][1]=t;
        update(x);
        t=x;
        x=fa[x];
    }
}
void makeroot(int x)
{
    access(x);splay(x);rev[x]^=1;
}
void split(int u,int v)
{
    makeroot(v);access(u);splay(u);
}
void link(int u,int v)
{
    makeroot(u);fa[u]=v;
}
void cut(int u,int v)
{
    makeroot(u);access(v);splay(v);
    c[v][0]=fa[u]=0;
}
int main()
{
    n=read();int Q=read();
    for(int i=1;i<=n;i++)
        val[i]=sum[i]=mt[i]=size[i]=1;
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read();
        link(u,v);
    }
    char ch[10];
    while (Q--)
    {
        scanf("%s",ch);
        int u=read(),v=read();
        if (ch[0]=='+')
        {
            int c=read();
            split(u,v);cal(u,1,c);
        }
        else if (ch[0]=='-')
        {
            cut(u,v);
            u=read(),v=read();
            link(u,v);
        }
        else if (ch[0]=='*')
        {
            int c=read();
            split(u,v);cal(u,c,0);
        }
        else
        {
            split(u,v);
            printf("%d\n",sum[u]);
        }
    }
    return 0;
}