BZOJ4034: [HAOI2015]树上操作(DFS序+线段树+树上问题维护贡献值)

题意:传送门
题解:对于1操作而言,当这个点增加一个值时,那么对于它以及它的子树而言路径的贡献都会加上这个值,这个对于刚开始的值也同样是这个道理,对于2操作而言,它以及它的子树都加上一个值,那么对于某个子树内的点贡献增加了 ( d e p i − d e p x + 1 ) ∗ a = d e p i ∗ a + ( 1 − d e p x ) ∗ a (dep_i-dep_x+1)*a=dep_i*a+(1-dep_x)*a (depidepx+1)a=depia+(1depx)a,对于后面这个 ( 1 − d e p x ) ∗ a (1-dep_x)*a (1depx)a可以用操作一中的方法来维护,对于前面这个可以只维护 a a a这个值,最后答案用 d e p i dep_i depi来进行相乘并加以维护。
附上DFS序代码:

void dfs(int u,int f)
{
    in[u]=++cnt;
    for(int i=head[u];~i;i=edges[i].next){
        int v=edges[i].v;
        if(v==f)continue;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
    out[u]=cnt;
}

附上代码:

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
int n,m,a[maxn],u,v,opt,id,x,cnt,in[maxn],out[maxn],dep[maxn];
struct edge{int u,v,next;}edges[maxn<<1];
int head[maxn],tot;
void init(){memset(head,-1,sizeof(head));tot=0;}
void add_edges(int u,int v)
{
    edges[tot].u=u;edges[tot].v=v;edges[tot].next=head[u];head[u]=tot++;
    edges[tot].u=v;edges[tot].v=u;edges[tot].next=head[v];head[v]=tot++;
}
void dfs(int u,int f)
{
    in[u]=++cnt;
    for(int i=head[u];~i;i=edges[i].next){
        int v=edges[i].v;
        if(v==f)continue;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
    out[u]=cnt;
}
struct data{
    int l,r,add,mul;
}tree[maxn<<2];
void build(int k,int s,int t)
{
    tree[k].l=s;tree[k].r=t;
    if(s==t)return ;
    int mid=(s+t)>>1;
    build(k<<1,s,mid);
    build(k<<1|1,mid+1,t);
}
void push_down(int k)
{
    tree[k<<1].add+=tree[k].add;
    tree[k<<1|1].add+=tree[k].add;
    tree[k<<1].mul+=tree[k].mul;
    tree[k<<1|1].mul+=tree[k].mul;
    tree[k].add=tree[k].mul=0;
}
void update(int k,int s,int t,int x,int y)
{
    int l=tree[k].l,r=tree[k].r;
    if(l==s&&r==t){
        tree[k].add+=x;tree[k].mul+=y;
        return ;
    }
    int mid=(l+r)>>1;
    push_down(k);
    if(t<=mid)update(k<<1,s,t,x,y);
    else if(s>mid)update(k<<1|1,s,t,x,y);
    else{
        update(k<<1,s,mid,x,y);
        update(k<<1|1,mid+1,t,x,y);
    }
}
int query(int k,int x,int s,int t)
{
    int l=tree[k].l,r=tree[k].r;
    if(l==s&&r==t){
        return dep[x]*tree[k].mul+tree[k].add;
    }
    push_down(k);
    int mid=(l+r)>>1;
    if(t<=mid)return query(k<<1,x,s,t);
    else if(s>mid)return query(k<<1|1,x,s,t);
    else return query(k<<1,x,s,mid)+query(k<<1,x,mid+1,t);
}
int32_t main()
{
    scanf("%lld%lld",&n,&m);
    init();
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
    }
    for(int i=1;i<n;i++){
        scanf("%lld%lld",&u,&v);
        add_edges(u,v);
    }
    dep[1]=1;
    dfs(1,-1);
    build(1,1,n);
    for(int i=1;i<=n;i++){
        update(1,in[i],out[i],a[i],0);
    }
    for(int i=1;i<=m;i++){
        scanf("%lld",&opt);
        if(opt==1){
            scanf("%lld%lld",&id,&x);
            update(1,in[id],out[id],x,0);
        }else if(opt==2){
            scanf("%lld%lld",&id,&x);
            update(1,in[id],out[id],(1-dep[id])*x,x);
        }else{
            scanf("%lld",&x);
            printf("%lld\n",query(1,x,in[x],in[x]));
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值