【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 intusing 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;}