【BZOJ4034】【HAOI2015】树上操作

题目请自行查阅传送门。

典型的树剖题，线段树维护操作，记一下子树在线段树内范围即可。

时间复杂度：\( O(m \log^{2} n) \)

#include <stdio.h>
#define MN 100005
#define Mn (1<<17)
#define ls (k<<1)
#define rs (k<<1|1)
#define mid (l+r>>1)
#define ll long long
ll sum[Mn<<],mark[Mn<<];
int val[MN],siz[MN],dep[MN],fa[MN],top[MN],son[MN],head[MN],l[MN],r[MN],n,q,cnt,dfsn,root=;
struct tree{int to,nxt;}edge[MN<<];
inline int in(){
    int x=,f=;
    char ch=getchar();
    while (ch<''||ch>'') f=ch=='-'?-:,ch=getchar();
    while (ch>=''&&ch<='') x=(x<<)+(x<<)+ch-'',ch=getchar();
    return x*f;
}
inline void ins(int x,int y){edge[++cnt].to=y,edge[cnt].nxt=head[x],head[x]=cnt;}
inline void dfs1(int u,int f,int d){
    siz[u]=,dep[u]=d,fa[u]=f;
    for (register int i=head[u]; i; i=edge[i].nxt)
        if (edge[i].to!=f){
            dfs1(edge[i].to,u,d+);siz[u]+=siz[edge[i].to];
            if (siz[edge[i].to]>siz[son[u]]) son[u]=edge[i].to;
        }
}
inline void dfs2(int u,int tp){
    top[u]=tp,l[u]=(++dfsn);if (son[u]) dfs2(son[u],tp);
    for (register int i=head[u]; i; i=edge[i].nxt)
        if (edge[i].to!=son[u]&&edge[i].to!=fa[u]) dfs2(edge[i].to,edge[i].to); r[u]=dfsn;
}
inline void pushdown(int k,int l,int r){
    if (l==r||!mark[k]) return;register int x=r-l+;
    mark[ls]+=mark[k],mark[rs]+=mark[k];
    sum[ls]+=(x-(x>>))*mark[k];sum[rs]+=(x>>)*mark[k];mark[k]=;
}
inline void combine(int k){sum[k]=sum[ls]+sum[rs];}
inline void A(int l,int r,int a,int b,int k,int ad){
    if (a<=l&&r<=b){mark[k]+=ad;sum[k]+=1ll*ad*(r-l+);return;}
    pushdown(k,l,r);if (a<=mid) A(l,mid,a,b,ls,ad);
    if (b>mid) A(mid+,r,a,b,rs,ad);combine(k);
}
inline ll Q(int l,int r,int a,int b,int k){
    if (l==a&&r==b) return sum[k];pushdown(k,l,r);
    if (b<=mid) return Q(l,mid,a,b,ls);
    if (a>mid) return Q(mid+,r,a,b,rs);
    return Q(l,mid,a,mid,ls)+Q(mid+,r,mid+,b,rs);
}
inline ll query(int x){
    register ll res=;
    while (top[x]!=root) res+=Q(,n,l[top[x]],l[x],),x=fa[top[x]];
    return res+=Q(,n,l[root],l[x],);
}
void init(){
    n=in(),q=in();for (int i=; i<=n; ++i) val[i]=in();
    for (register int i=; i<n; ++i){
        register int x=in(),y=in();
        ins(x,y); ins(y,x);
    }
    dfs1(root,,);dfs2(root,root);
    for (register int i=; i<=n; ++i) A(,n,l[i],l[i],,val[i]);
}
void solve(){
    while(q--){
        register int op=in(),x=in(),ad;
        if (op^){
            ad=in();if (op&) A(,n,l[x],l[x],,ad);
            else A(,n,l[x],r[x],,ad);
         }else printf("%lld\n",query(x));
    }
}
int main(){init(); solve(); return ;}