Codechef June Challenge 2017 #Persistent oak -- 树链剖分+主席树

传送门

每个点维护一下还能承受多少重量，线段树维护区间最小值就可以了。
如果一个子树全部折断，可以直接把初始版本的主席树接过来。

代码：

#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void Read(int& x){
    char c=nc();
    for(;c<'0'||c>'9';c=nc());
    for(x=0;c>='0'&&c<='9';x=(x<<3)+(x<<1)+c-48,c=nc());
}
char ss[30];
int Len;
inline void Print(int x){
    if(!x){
        puts("0");
        return;
    }
    for(Len=0;x;x/=10)ss[++Len]=x%10;
    while(Len)putchar(ss[Len--]+48);
    putchar('\n');
}
#define N 100010
#define M 100000000
int cl[M],cr[M],cw[M],cp[M];
int top[N],s[N],son[N],f[N],w[N],w2[N];
int i,j,k,n,m,p,Rt[N],t;
int T,x,y,a[N],R[N],num,cnt,b[N<<2];
vector<int>g[N];
inline void Dfs(int x){
    s[x]=1;son[x]=0;
    for(int i=0;i<g[x].size();i++){
        Dfs(g[x][i]);
        s[x]+=s[g[x][i]];
        if(s[g[x][i]]>s[son[x]])son[x]=g[x][i];
    }
}
inline void Dfs2(int x,int y){
    top[x]=y;w[x]=++num;w2[num]=x;
    if(son[x])Dfs2(son[x],y);
    for(int i=0;i<g[x].size();i++)
    if(g[x][i]!=son[x])Dfs2(g[x][i],g[x][i]);
    R[x]=num;
}
inline void Build(int& x,int y,int l,int r){
    b[y]=x=++cnt;
    cl[x]=cr[x]=cp[x]=0;
    if(l==r){
        cw[x]=a[w2[l]];
        return;
    }
    int Mid=l+r>>1;
    Build(cl[x],y<<1,l,Mid);Build(cr[x],y<<1|1,Mid+1,r);
    cw[x]=min(cw[cl[x]],cw[cr[x]]);
}
inline void Update_add(int& x,int z){
    int y=x;
    x=++cnt;
    cl[x]=cl[y];cr[x]=cr[y];cw[x]=cw[y]+z;cp[x]=cp[y]+z;
}
inline void Down(int x){
    if(cp[x]){
        Update_add(cl[x],cp[x]);
        Update_add(cr[x],cp[x]);
        cp[x]=0;
    }
}
inline void Update(int& x,int y,int l,int r,int L,int R,int z){
    if(l>R||r<L)return;
    Down(x);
    if(l>=L&&r<=R){
        Update_add(x,z);
        return;
    }
    x=++cnt;
    cl[x]=cl[y];cr[x]=cr[y];cp[x]=cp[y];cw[x]=cw[y];
    int Mid=l+r>>1;
    Update(cl[x],cl[y],l,Mid,L,R,z);Update(cr[x],cr[y],Mid+1,r,L,R,z);
    cw[x]=min(cw[cl[x]],cw[cr[x]]);
}
inline void Update_tree(int &x,int y,int z){
    while(y){
        Update(x,x,2,n,w[top[y]],w[y],z);
        y=f[top[y]];
    }
}
inline int Find(int x,int l,int r,int L,int R){
    if(l>R||r<L||cw[x]>=0)return -1;
    if(l==r)return w2[l];
    Down(x);
    int Mid=l+r>>1;
    if(l>=L&&r<=R){
        if(cw[cr[x]]<0)return Find(cr[x],Mid+1,r,L,R);
        return Find(cl[x],l,Mid,L,R);
    }
    int t=Find(cr[x],Mid+1,r,L,R);
    if(t!=-1)return t;
    return Find(cl[x],l,Mid,L,R);
}
inline int Find_tree(int x,int y){
    while(y){
        int t=Find(x,2,n,w[top[y]],w[y]);
        if(t!=-1)return t;
        y=f[top[y]];
    }
    return -1;
}
inline void Update2(int &x,int y,int z,int l,int r,int L,int R){
    if(l>R||r<L)return;
    Down(x);
    if(l>=L&&r<=R){
        x=b[z];
        return;
    }
    x=++cnt;
    cl[x]=cl[y];cr[x]=cr[y];cp[x]=0;cw[x]=cw[y];
    int Mid=l+r>>1;
    Update2(cl[x],cl[y],z<<1,l,Mid,L,R);Update2(cr[x],cr[y],z<<1|1,Mid+1,r,L,R);
    cw[x]=min(cw[cl[x]],cw[cr[x]]);
}
inline int Query(int x,int l,int r,int y){
    if(l==r)return cw[x];
    Down(x);
    int Mid=l+r>>1;
    if(y<=Mid)return Query(cl[x],l,Mid,y);
    return Query(cr[x],Mid+1,r,y);
}
int main(){
    Read(T);
    while(T--){
        Read(n);Read(m);n++;
        for(i=1;i<=n;i++)g[i].clear();
        for(i=2;i<=n;i++)Read(f[i]),Read(a[i]),g[++f[i]].push_back(i);
        num=cnt=0;Dfs(1);Dfs2(1,1);
        Build(Rt[0],1,2,n);
        for(i=1;i<=m;i++){
            Read(p);Read(k);Read(x);x++;
            Rt[i]=Rt[p];
            if(k==1){
                Read(y);
                Update_tree(Rt[i],x,-y);
                t=Find_tree(Rt[i],x);
                if(t==-1)puts("0");else{
                    Print(t-1);
                    Update_tree(Rt[i],x,y);
                    Update_tree(Rt[i],t,a[t]-Query(Rt[i],2,n,w[t]));
                    Update2(Rt[i],Rt[i],1,2,n,w[t],R[t]);
                }
            }else{
                t=a[x]-Query(Rt[i],2,n,w[x]);
                Print(t);
                if(t)Update_tree(Rt[i],x,t);
                Update2(Rt[i],Rt[i],1,2,n,w[x],R[x]);
            }
        }
    }
    return 0;
}