bzoj 4034(DFS序+线段树)

Posted 2020-08-14 AC菜鸟机

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

 有一棵点数为 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 = 800005;
struct Edge
{
    int v,next;
} edge[N<<1];
int head[N],tot,cnt;
int val[N];
int in[N],out[N];
LL tree[N<<2],lazy[N<<2],seg[N];
int flag[N<<2];
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,-1,sizeof(head));
    memset(val,0,sizeof(val));
    memset(lazy,0,sizeof(lazy));
    memset(io,0,sizeof(io));
    memset(seg,0,sizeof(seg));
    tot = cnt = 0;
}
void dfs(int u,int fa)
{
    seg[in[u] = ++cnt] = (LL)val[u];
    io[cnt] = 1;
    for(int i=head[u]; i!=-1; i=edge[i].next)
    {
        if(edge[i].v != fa)
        {
            dfs(edge[i].v,u);
        }
    }
    seg[out[u] = ++cnt] = (LL)(-val[u]);
    io[cnt] = -1;
}

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