bzoj 4034(DFS序+线段树)

这个题多了一个操作难度直线上升,看完题解才会写

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个
操作，分为三种：
操作 1 ：把某个节点 x 的点权增加 a 。
操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

按照题意:记录其DFS序,然后进栈是正,出栈为负,利用线段树进行更新

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <vector>
#include <stack>
using namespace std;
typedef long long LL;
const int N = ;
struct Edge
{
    int v,next;
} edge[N<<];
int head[N],tot,cnt;
int val[N];
int in[N],out[N];
LL tree[N<<],lazy[N<<],seg[N];
int flag[N<<];
int io[N];
void addEdge(int u,int v)
{
    edge[tot].v = v,edge[tot].next = head[u],head[u] = tot++;
}
void init()
{
    memset(head,-,sizeof(head));
    memset(val,,sizeof(val));
    memset(lazy,,sizeof(lazy));
    memset(io,,sizeof(io));
    memset(seg,,sizeof(seg));
    tot = cnt = ;
}
void dfs(int u,int fa)
{
    seg[in[u] = ++cnt] = (LL)val[u];
    io[cnt] = ;
    for(int i=head[u]; i!=-; i=edge[i].next)
    {
        if(edge[i].v != fa)
        {
            dfs(edge[i].v,u);
        }
    }
    seg[out[u] = ++cnt] = (LL)(-val[u]);
    io[cnt] = -;
}

void pushup(int idx)
{
    tree[idx] = tree[idx<<]+tree[idx<<|];
    return;
}
void pushdown(int idx)
{
    if(lazy[idx])
    {
        tree[idx << ] +=  flag[idx<<]*lazy[idx];
        tree[idx <<  | ] += flag[idx<<|]*lazy[idx];
        lazy[idx << ] += lazy[idx];
        lazy[idx <<  | ] += lazy[idx];
        lazy[idx] = ;
    }
    return;
}
void build(int l,int r,int idx)
{
    if(l==r)
    {
        tree[idx] = seg[l];
        if(io[l]>) flag[idx] = ;
        else flag[idx] = -;
        return;
    }
    int mid = (l+r)>>;
    build(l,mid,idx<<);
    build(mid+,r,idx<<|);
    pushup(idx);
    flag[idx] = flag[idx<<] + flag[idx<<|];
}
void update(int l,int r,int L,int R,int idx,int val)
{
    if(l>=L&&r<=R)
    {
        tree[idx] =tree[idx] + (LL)flag[idx]*val;
        lazy[idx] =lazy[idx] + (LL)val;
        return;
    }
    int mid = (l+r)>>;
    pushdown(idx);
    if(mid>=L) update(l,mid,L,R,idx<<,val);
    if(mid<R) update(mid+,r,L,R,idx<<|,val);
    pushup(idx);
}
LL query(int l,int r,int L,int R,int idx){
    if(l>=L&&r<=R)
    {
        return tree[idx];
    }
    int mid = (l+r)>>;
    LL sum = ;
    pushdown(idx);
    if(mid>=L) sum+=query(l,mid,L,R,idx<<);
    if(mid<R) sum+=query(mid+,r,L,R,idx<<|);
    return sum;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        for(int i=; i<=n; i++)
        {
            scanf("%d",&val[i]);
        }
        for(int i=; i<n; i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            addEdge(u,v);
            addEdge(v,u);
        }
        dfs(,);
        build(,*n,);
        while(m--)
        {
            int op,x,y;
            scanf("%d",&op);
            if(op==)
            {
                scanf("%d%d",&x,&y);
                update(,*n,in[x],in[x],,y);
                update(,*n,out[x],out[x],,y);
            }else if(op==){
                scanf("%d%d",&x,&y);
                update(,*n,in[x],out[x],,y);
            }else{
                scanf("%d",&x);
                LL sum = query(,*n,,in[x],);
                printf("%lld\n",sum);
            }
        }
    }
}