[HAOI2015]树上操作

问题描述

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

输入格式

第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数，表示树中节点的初始权值。接下来 N-1 行每行三个正整数 fr, to ， 表示该树中存在一条边 (fr, to) 。再接下来 M 行，每行分别表示一次操作。其中第一个数表示该操作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。

输出格式

对于每个询问操作，输出该询问的答案。答案之间用换行隔开。

样例输入

5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3

样例输出

6
9
13

数据范围

对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不会超过 10^6 。

---

题解

似乎这道题可以用树链剖分，但本蒟蒻目前还不会，就用的 DFS序+线段树。
DFS预处理出进入、离开每个点的时间戳。这两个时间戳就是这个点的儿子的时间戳的左、右区间。
对于操作一，显然改变一个点的值只会影响到它的儿子，此操作可由DFS序改为区间修改。所以可以用线段树维护。
操作二较为恶心，因为子树内不同的点到根的距离变化量是不同的。某一点的变化量$\Delta$j=(dep[j]-dep[x])*a，dep[]记录深度。这玩意儿不好维护，不妨将其拆解为（dep[j]*a-dep[x]*a）。当x、a为定值时，后半部分为定值，用操作一的方法维护。再用线段树维护前半部分的 a值的修改。
操作三显然是点查询，输出 操作二前半部分乘深度再加上操作一的值，就行了。
具体见代码

---

代码

又是动态开点。。。

#include <stdio.h>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#define ll long long
using namespace std;
const ll Q=100005;
ll in[Q],out[Q],nn[Q<<1],e[Q<<1],last[Q],tot=0,n,cnt=0,dep[Q],a[Q],ls[Q<<4],rs[Q<<4],lazy[Q<<4],lan[Q<<4];
bool ch[Q];
void search(ll x,ll now)
{
    ll y,t;
    ch[x]=true;
    in[x]=++cnt;
    dep[x]=now;
    for(t=last[x];t;t=nn[t])
    {
        y=e[t];
        if(ch[y])continue;
        search(y,now+1);
    }
    out[x]=cnt;
}
void add(ll x,ll y)
{
    e[++tot]=y;
    nn[tot]=last[x];
    last[x]=tot;
}
void xiu(ll now,ll l,ll r,ll x,ll y,ll v)
{
    if(x<=l&&y>=r)
    {
        lan[now]+=v;
        return;
    }
    if(lan[now]){
        if(!ls[now])ls[now]=++tot;
        if(!rs[now])rs[now]=++tot;
        lan[ls[now]]+=lan[now];
        lan[rs[now]]+=lan[now];
        lan[now]=0;
    }
    ll mid=(l+r)>>1;
    if(x<=mid){
        if(!ls[now])ls[now]=++tot;
        xiu(ls[now],l,mid,x,y,v);
    }
    if(y>mid){
        if(!rs[now])rs[now]=++tot;
        xiu(rs[now],mid+1,r,x,y,v);
    }
}
ll gs(ll now,ll l,ll r,ll x)
{
    if(now==0)return 0;
    if(l==r)return lan[now];
    if(lan[now]){
        if(!ls[now])ls[now]=++tot;
        if(!rs[now])rs[now]=++tot;
        lan[ls[now]]+=lan[now];
        lan[rs[now]]+=lan[now];
        lan[now]=0;
    }
    ll mid=(l+r)>>1;
    if(x<=mid)return gs(ls[now],l,mid,x);
    return gs(rs[now],mid+1,r,x);
}
int main()
{
    ll i,x,y,m;
    scanf("%lld%lld",&n,&m);
    for(i=1;i<=n;i++)scanf("%lld",&a[i]);
    for(i=1;i<n;i++)
    {
        scanf("%lld%lld",&x,&y);
        add(x,y),add(y,x);
    }
    tot=2;
    search(1,1);
    for(i=1;i<=n;i++)
        xiu(1,1,n,in[i],out[i],a[i]);
    while(m--)
    {
        scanf("%lld%lld",&i,&x);
        if(i==1)scanf("%lld",&y),xiu(1,1,n,in[x],out[x],y);
        else if(i==2)scanf("%lld",&y),xiu(1,1,n,in[x],out[x],y*(1-dep[x])),xiu(2,1,n,in[x],out[x],y);
        else printf("%lld\n",gs(1,1,n,in[x])+gs(2,1,n,in[x])*dep[x]);
    }
    return 0;
}