bzoj 2631 tree

2631: tree

Time Limit: 30 Sec   Memory Limit: 128 MB
Submit: 4429   Solved: 1488
[ Submit][ Status][ Discuss]

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

Source

【分析】

一种想rigel的冲动...

唉pushdown的时候乘法标记写成 if(mul>1) 再操作了...殊不知还有mul==0的情况23333

改的天昏地暗...终于在二分提交中试出了错误...

【代码】

//bzoj 2631 Tree
#include<iostream>
#include<cstring>
#include<cstdio>
#define p 51061
#define ll unsigned int
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=100005;
char s[2];
int n,m,top;
int f[mxn],ch[mxn][2],st[mxn],size[mxn];
ll sum[mxn],add[mxn],mul[mxn],rev[mxn],key[mxn];
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<<1)+(x<<3)+ch-'0',ch=getchar();
    return x*f;
}
inline bool isroot(int x)
{
    return (ch[f[x]][0]!=x && ch[f[x]][1]!=x);
}
inline int get(int x)
{
    if(ch[f[x]][0]==x) return 0;return 1;
}
inline void update(int x)
{
    size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
    sum[x]=(sum[ch[x][0]]+sum[ch[x][1]]+key[x])%p;
}
inline void ope_mul(int x,ll mu)
{
    add[x]=add[x]*mu%p;
    mul[x]=mul[x]*mu%p;
    key[x]=key[x]*mu%p;
    sum[x]=sum[x]*mu%p;
}
inline void ope_add(int x,ll c)
{
    add[x]=(add[x]+c)%p;
    key[x]=(key[x]+c)%p;
    sum[x]=(sum[x]+c*size[x]%p)%p;
}
inline void pushdown(int x)
{
    if(rev[x])
    {
        rev[ch[x][0]]^=1,rev[ch[x][1]]^=1;
        swap(ch[x][0],ch[x][1]);
    }
    if(mul[x]!=1)
    {
        if(ch[x][0]) ope_mul(ch[x][0],mul[x]);
        if(ch[x][1]) ope_mul(ch[x][1],mul[x]);
    }
    if(add[x])
    {
        if(ch[x][0]) ope_add(ch[x][0],add[x]);
        if(ch[x][1]) ope_add(ch[x][1],add[x]);
    }
    update(x);
    rev[x]=add[x]=0,mul[x]=1;
}
inline void rotate(int x)   //
{
    pushdown(x);
    int fa=f[x],fafa=f[fa],which=get(x);
    if(!isroot(fa)) ch[fafa][ch[fafa][1]==fa]=x;
    f[x]=fafa;
    ch[fa][which]=ch[x][which^1],f[ch[fa][which]]=fa;
    ch[x][which^1]=fa,f[fa]=x;
    update(fa),update(x);
}
inline void splay(int x)   //
{
    st[top=1]=x;
    for(int i=x;!isroot(i);i=f[i]) st[++top]=f[i];
    for(int i=top;i;i--) pushdown(st[i]);
    for(int fa;!isroot(x);rotate(x))
      if(!isroot(fa=f[x])) rotate(get(x)==get(fa)?fa:x);
}
inline void access(int x)  //
{
    for(int y=0;x;y=x,x=f[x])
      splay(x),ch[x][1]=y;
}  
inline void makeroot(int x)  //
{
    access(x),splay(x),rev[x]^=1;
}
inline void link(int x,int y)  //
{
    makeroot(x);
    f[x]=y,splay(x);
}
inline void cut(int x,int y)  //
{
    makeroot(x);
    access(y),splay(y);
    f[x]=ch[y][0]=0;
}
int main()
{
    int i,j,u,v,c,uu,vv;
    n=read(),m=read();
    fo(i,1,n) key[i]=sum[i]=mul[i]=1;
    fo(i,2,n)
    {
        u=read(),v=read();
        link(u,v);
    }
    while(m--)
    {
        scanf("%s",s);
        if(s[0]=='+')
        {
            u=read(),v=read(),c=read()%p;
            makeroot(u),access(v),splay(v);
            ope_add(v,c);
        }
        else if(s[0]=='-')
        {
            u=read(),v=read(),uu=read(),vv=read();
            cut(u,v),link(uu,vv);
        }
        else if(s[0]=='*')
        {
            u=read(),v=read(),c=read()%p;
            makeroot(u),access(v),splay(v);
            ope_mul(v,c);
        }
        else
        {
            u=read(),v=read();
            makeroot(u),access(v),splay(v);
            printf("%d\n",sum[v]);
        }
    }
    return 0;
}