BZOJ4034_树上操作_KEY

题目传送门

这道题可以树链剖分+线段树。

其他操作模板，第二个操作只需要将x~x+size[x]-1区间加值即可。

code：

#include <cstdio>
#include <cstring>
using namespace std;

int read(){
    char c;while(c=getchar(),(c<''||c>'')&&c!='-');
    int x=,y=;c=='-'?y=-:x=c-'';
    while(c=getchar(),c>=''&&c<='')x=x*+c-'';
    return x*y;
}

const int Maxn=;

int N,Q,A[Maxn];
int head[Maxn],nxt[Maxn<<],to[Maxn<<],Cnt;
void Add(int x,int y){
    to[Cnt]=y;
    nxt[Cnt]=head[x];
    head[x]=Cnt;
    Cnt++;
}

int Son[Maxn],Size[Maxn],f[Maxn];
int Dep[Maxn],Dfn[Maxn],Top[Maxn];
int C;
void Find(int Now,int Deep,int Bef){

    f[Now]=Bef,Dep[Now]=Deep,Size[Now]=;
        for(int i=head[Now];i!=-;i=nxt[i]){
            if(to[i]==f[Now])continue;
            Find(to[i],Deep+,Now);
            Size[Now]+=Size[to[i]];
            if(Size[to[i]]>Size[Son[Now]])
                Son[Now]=to[i];
        }
    return ;
}
void Fs(int Now,int Tp){
    Dfn[Now]=++C,Top[Now]=Tp;
    if(Son[Now])Fs(Son[Now],Tp);
        for(int i=head[Now];i!=-;i=nxt[i]){
            if(to[i]==Son[Now]||to[i]==f[Now])continue;
            Fs(to[i],to[i]);
        }
    return ;
}

long long Seg[Maxn<<],Ade[Maxn<<];
void Up(int x){Seg[x]=Seg[x<<]+Seg[x<<|];}
void Down(int x,long long l,long long r){
    if(!Ade[x])return ;
    Seg[x<<]+=Ade[x]*l;
    Seg[x<<|]+=Ade[x]*r;
    Ade[x<<]+=Ade[x];
    Ade[x<<|]+=Ade[x];
    Ade[x]=;
}
void Updata(int Node,int L,int R,int Ul,int Ur,long long Val){
    if(Ul<=L&&Ur>=R){
        Seg[Node]+=Val*(R-L+);
        Ade[Node]+=Val;
        return ;
    }
    int Mid=L+R>>;
    Down(Node,Mid-L+,R-Mid);
    if(Mid>=Ul)Updata(Node<<,L,Mid,Ul,Ur,Val);
    if(Mid< Ur)Updata(Node<<|,Mid+,R,Ul,Ur,Val);
    Up(Node);
}
long long Query(int Node,int L,int R,int Ql,int Qr){
    if(Ql<=L&&Qr>=R)return Seg[Node];
    int Mid=L+R>>;
    Down(Node,Mid-L+,R-Mid);
    long long Ans=;
    if(Mid>=Ql)Ans+=Query(Node<<,L,Mid,Ql,Qr);
    if(Mid< Qr)Ans+=Query(Node<<|,Mid+,R,Ql,Qr);
    return Ans;
}
long long G(int Node){
    long long Ans=;
        while(Top[Node]){
            Ans+=Query(,,N,Dfn[Top[Node]],Dfn[Node]);
            Node=f[Top[Node]];
        }
    return Ans;
}

int main()
{
    memset(head,-,sizeof head);
    N=read(),Q=read();
        for(int i=;i<=N;i++)A[i]=read();
        for(int i=;i<N;i++){
            int x=read(),y=read();
            Add(x,y),Add(y,x);
        }
    Find(,,),Fs(,);
        for(int i=;i<=N;i++)
            Updata(,,N,Dfn[i],Dfn[i],A[i]);
        for(int i=;i<=Q;i++){
            int Type=read(),x=read();
            if(Type==)Updata(,,N,Dfn[x],Dfn[x],read());
            if(Type==)Updata(,,N,Dfn[x],Dfn[x]+Size[x]-,read());
            if(Type==)printf("%lld\n",G(x));
        }
    return ;
}