BZOJ2631: tree

2631: tree

Time Limit: 30 Sec  Memory Limit: 128 MB
Submit: 1486  Solved: 515
[Submit][Status]

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

题解：

在pushdown上加一些东西就可以。坑爹的题目居然还要用unsigned int

我惊呆了，long long 居然T了，差这么多。。。

代码：

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<string>
#define inf 1000000000
#define maxn 100000+100
#define maxm 500+100
#define eps 1e-10
#define ll long long
#define pa pair<int,int>
#define mod 51061
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=10*x+ch-'0';ch=getchar();}
    return x*f;
}
struct edge{int go,next;}e[2*maxn];
unsigned int  n,m,tot,head[maxn],fa[maxn],c[maxn][2],s[maxn];
unsigned int sta[maxn],add[maxn],mul[maxn],sum[maxn],v[maxn];
bool rev[maxn];
inline void insert(int x,int y)
{
    e[++tot].go=y;e[tot].next=head[x];head[x]=tot;
    e[++tot].go=x;e[tot].next=head[y];head[y]=tot;
}
inline bool isroot(int x)
{
    return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;
}
inline void pushup(int x)
{
    s[x]=s[c[x][0]]+s[c[x][1]]+1;
    sum[x]=(sum[c[x][0]]+sum[c[x][1]]+v[x])%mod;
}
inline void rotate(int x)
{
    int y=fa[x],z=fa[y],l=(c[y][1]==x),r=l^1;
    if(!isroot(y))c[z][c[z][1]==y]=x;
    fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
    c[y][l]=c[x][r];c[x][r]=y;
    pushup(y);pushup(x);
}
inline void update(int x,int y,int z)
{
    if(!x)return;
    add[x]=(add[x]*z+y)%mod;mul[x]=(mul[x]*z)%mod;
    sum[x]=(sum[x]*z+s[x]*y)%mod;v[x]=(v[x]*z+y)%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]);
    }
    update(l,add[x],mul[x]);update(r,add[x],mul[x]);
    add[x]=0;mul[x]=1;
}
inline void splay(int x)
{
    int top=0;sta[++top]=x;
    for(int y=x;!isroot(y);y=fa[y])sta[++top]=fa[y];
    for(;top;)pushdown(sta[top--]);
    while(!isroot(x))
    {
        int y=fa[x],z=fa[y];
        if(!isroot(y))
         {
             if(c[z][0]==y^c[y][0]==x)rotate(x);else rotate(y);
         }
        rotate(x); 
    }
    
}
inline void access(int x)
{
    for(int y=0;x;x=fa[x])
    {
        splay(x);c[x][1]=y;pushup(x);y=x;
    }
}
inline void makeroot(int x)
{
    access(x);splay(x);rev[x]^=1;
}
inline int find(int x)
{
    access(x);splay(x);
    while(c[x][0])x=c[x][0];
    return x;
}
inline void link(int x,int y)
{
    makeroot(x);fa[x]=y;splay(x);
}
inline void cut(int x,int y)
{
    makeroot(x);access(y);splay(y);c[y][0]=fa[x]=0;
}
inline void split(int x,int y)
{
    makeroot(x);access(y);splay(y);
}
int main()
{
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
    n=read();m=read();
    int x,y,z,top=0;char ch[10];
    for(int i=1;i<n;i++)x=read(),y=read(),insert(x,y);
    for(int i=1;i<=n;i++)s[i]=sum[i]=v[i]=mul[i]=1,add[i]=0;
    sta[++top]=1;
    for(int i=1;i<=top;i++)
     for(int k=sta[i],j=head[k];j;j=e[j].next)
      if(e[j].go!=fa[k])
       {
           fa[y=e[j].go]=k;sta[++top]=y;
       }  
    while(m--)
    {
        scanf("%s",ch);
        switch(ch[0])
        {
            case '-':{cut(read(),read());link(read(),read());}break;
            case '/':{x=read();y=read();split(x,y);printf("%d
",sum[y]);}break;
            case '+':{x=read();y=read();z=read();split(x,y);update(y,z,1);}break;
            case '*':{x=read();y=read();z=read();split(x,y);update(y,0,z);}break;
        }
    }
    return 0;
}