BZOJ 2631 tree

title: ‘BZOJ 2631 tree’
categories: BZOJ
date: 2016-1-14 22:32:00
tags: [LCT]

---

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.txt
3 2
1 2
2 3
* 1 3 4
\/ 1 1

output.txt
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

Solution

本题主要是tag的维护比较麻烦。具体的操作可以看代码。
有一个坑点是要开unsigned int，据黄学长说开long long会超时，于是开unsigned int就好了。。。我被这个坑了一晚上没发现，以为自己模板打挂了，好蛋疼啊QAQ……

Code

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

#define maxn 100000+5
#define mod 51061

using namespace std;

typedef unsigned int ll;

struct Splay_Tree{
    int ch[2];
    int flip,sz,pf,fa;
    ll val,pl,mul,tag,sum;
}tr[maxn];

int stack[maxn];
int n,m,top;

inline int in(){
    int x=0;
    char ch=getchar();      
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}

void Calc(int k,int pl,int mul){
    tr[k].sum=(tr[k].sum*mul+tr[k].sz*pl)%mod;
    tr[k].val=(tr[k].val*mul+pl)%mod;
    tr[k].pl=(tr[k].pl*mul+pl)%mod;
    tr[k].mul=(tr[k].mul*mul)%mod;
}

void Pushdown(int k){
    int l=tr[k].ch[0],r=tr[k].ch[1];
    if(tr[k].flip){
        tr[k].flip^=1;
        tr[l].flip^=1; tr[r].flip^=1;
        swap(tr[k].ch[0],tr[k].ch[1]);
    }
    if(tr[k].pl || tr[k].mul!=1){
        if(l) Calc(l,tr[k].pl,tr[k].mul);
        if(r) Calc(r,tr[k].pl,tr[k].mul);
        tr[k].pl=0; tr[k].mul=1;
    }
}

void Update(int k){
    int l=tr[k].ch[0],r=tr[k].ch[1];
    tr[k].sz=(tr[l].sz+tr[r].sz+1)%mod;
    tr[k].sum=(tr[l].sum+tr[r].sum+tr[k].val)%mod;
}

void Rotate(int &k,int d){
    int t=tr[k].ch[d^1];
    tr[t].pf=tr[k].pf; tr[t].fa=tr[k].fa;
    if(tr[t].ch[d]) tr[tr[t].ch[d]].fa=k;
    tr[k].pf=0; tr[k].fa=t;
    tr[k].ch[d^1]=tr[t].ch[d]; tr[t].ch[d]=k;   
    Update(k); Update(t); k=t;
}

void Splay(int &k,int x){
    int d1=(tr[tr[k].ch[0]].sz<x?1:0),t=tr[k].ch[d1];
    if(d1==1) x-=tr[tr[k].ch[0]].sz+1;
    if(x){
        int d2=(tr[tr[t].ch[0]].sz<x?1:0);
        if(d2==1) x-=tr[tr[t].ch[0]].sz+1;
        if(x){
            Splay(tr[t].ch[d2],x);
            if(d1==d2) Rotate(k,d1^1);
            else Rotate(tr[k].ch[d1],d1);   
        }
        Rotate(k,d1^1);
    }
}

int rank(int k,int &rt){
    int res=tr[tr[k].ch[0]].sz+1;
    stack[++top]=k;
    while(tr[k].fa){
        int p=tr[k].fa;
        stack[++top]=p;
        if(tr[p].ch[0]!=k)
            res+=tr[tr[p].ch[0]].sz+1;
        k=p;
    }
    rt=k;
    return res;
}

void Haha(int k){
    int rt,x=rank(k,rt);
    while(top) Pushdown(stack[top--]);
    x=rank(k,rt);
    Splay(rt,x);
}

void Cutright(int k){
    if(tr[k].ch[1]){
        tr[tr[k].ch[1]].pf=k;
        tr[tr[k].ch[1]].fa=0;
        tr[k].ch[1]=0;
        Update(k);
    }
}

void Access(int k){
    Haha(k); Cutright(k);
    while(tr[k].pf){
        int u=tr[k].pf;
        Haha(u); Cutright(u);
        tr[u].ch[1]=k;
        tr[k].pf=0; tr[k].fa=u;
        Update(u); k=u;
    }
}

void Beroot(int k){
    Access(k); Haha(k);
    tr[k].flip^=1;
}

int Find(int k){
    Access(k); Haha(k);
    while(tr[k].ch[0]) k=tr[k].ch[0];
    return k;
}

void Link(int x,int y){
    int rt1=Find(x),rt2=Find(y);
    if(rt1==rt2) return;
    Beroot(x); Access(y);
    tr[x].pf=y;
}

void Cut(int k,int p){
    Beroot(k); Access(p); Haha(p);
    tr[p].ch[0]=tr[k].pf=tr[k].fa=0;
    Update(p);
}

int main(){
    n=in(); m=in();
    for(int i=1;i<=n;i++)
        tr[i].sz=tr[i].sum=tr[i].val=tr[i].mul=1;
    for(int i=1;i<n;i++){
        int x,y;
        x=in(); y=in();
        Link(x,y);
    }
    char s[1];
    for(int i=1;i<=m;i++){
        int u,v,x,y,c;
        scanf("%s",s);
        if(s[0]=='+'){
            u=in(); v=in(); c=in();
            Beroot(v); Access(u); Haha(u);
            Calc(u,c,1);
        }
        else if(s[0]=='-'){
            u=in(); v=in(); x=in(); y=in();
            Cut(u,v); Link(x,y);
        }
        else if(s[0]=='*'){
            u=in(); v=in(); c=in();
            Beroot(v); Access(u); Haha(u);
            Calc(u,0,c);
        }
        else{
            u=in(); v=in();
            Beroot(v); Access(u); Haha(u);
            printf("%d\n",tr[u].sum);
        }
    }
    return 0;   
}