bzoj2836魔法树 树剖

题目大意:
初始点权为0
支持路径修改+子树求和
用树剖水过去
路径修改:区间修改即可
子树求和:区间求和(pos[x],pos[x]+size[x]-1)
PS:据说dfs序也可以做
把修改等价于x到根上路径的修改,然后考虑修改对于y的贡献.
只有x在y子树内才有贡献.且为
$W(x)*(dep(x)-dep(y)+1)$分离变量后用树状数组维护即可
参考 <<数据结构漫谈>>许昊然

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e5+10;
typedef long long ll;
int last[N],len=0,l[N],r[N],
ls[N<<2],rs[N<<2];ll tag[N<<2],sum[N<<2];
int dep[N],s[N],pos[N],bl[N],f[N],size=0,
n,q;
struct Edge{int to,next;Edge(int to=0,int next=0):to(to),next(next){}}e[N<<1];
void add_edge(int u,int v){e[++len]=Edge(v,last[u]);last[u]=len;}
void dfs1(int u) 
{
    s[u]=1;
    for(int i=last[u];i;i=e[i].next) {
        int id=e[i].to;
        if(id==f[u])continue;
        dep[id]=dep[u]+1;f[id]=u;
        dfs1(id);
        s[u]+=s[id];
    }
}
void dfs2(int u,int chain)
{
    bl[u]=chain;pos[u]=++size;l[u]=r[u]=size;
    int k=0;
    for(int i=last[u];i;i=e[i].next) {
        int id=e[i].to;
        if(dep[id]>dep[u]&&s[id]>s[k])k=id;
    }
    if(k==0)return;
    dfs2(k,chain);
    for(int i=last[u];i;i=e[i].next) {
        int id=e[i].to;
        if(dep[id]>dep[u]&&id!=k)dfs2(id,id);
    }r[u]=size;
}
void build(int k,int l,int r)
{
    ls[k]=l;rs[k]=r;sum[k]=tag[k]=0;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
}
void pushdown(int k)
{
    int len=(rs[k]-ls[k]+1);
    tag[k<<1]+=tag[k];
    tag[k<<1|1]+=tag[k];
    sum[k<<1]+=(len-(len>>1))*tag[k];
    sum[k<<1|1]+=(len>>1)*tag[k];
    tag[k]=0;
}
void change(int k,int ql,int qr,ll delta)
{
    if(ql==ls[k]&&qr==rs[k]){sum[k]+=delta*(rs[k]-ls[k]+1);tag[k]+=delta;return;}
    if(tag[k])pushdown(k);
    int mid=(ls[k]+rs[k])>>1;
    if(qr<=mid)change(k<<1,ql,qr,delta);
    else if(ql>mid)change(k<<1|1,ql,qr,delta);
    else {
        change(k<<1,ql,mid,delta);
        change(k<<1|1,mid+1,qr,delta);
    }
    sum[k]=sum[k<<1]+sum[k<<1|1];
}
ll querysum(int k,int ql,int qr)
{
    if(ls[k]==ql&&rs[k]==qr){return sum[k];}
    if(tag[k])pushdown(k);
    int mid=(ls[k]+rs[k])>>1;
    if(qr<=mid)return querysum(k<<1,ql,qr);
    else if(ql>mid)return querysum(k<<1|1,ql,qr);
    else return (querysum(k<<1,ql,mid)+querysum(k<<1|1,mid+1,qr));
}
void solveadd(int x,int y,ll val)
{
    while(bl[x]!=bl[y]) {
        if(dep[bl[x]]<dep[bl[y]])swap(x,y);
        change(1,pos[bl[x]],pos[x],val);
        x=f[bl[x]];
    }if(dep[x]>dep[y])swap(x,y);
    change(1,pos[x],pos[y],val);
}
int main()
{
    scanf("%d",&n);
    for(int from,to,i=1;i<n;i++) {
        scanf("%d%d",&from,&to);
        add_edge(from+1,to+1);add_edge(to+1,from+1);
    }
    dfs1(1);
    dfs2(1,1);
    build(1,1,n);
    scanf("%d",&q);
    for(int i=1;i<=q;i++) {
        char ch[10];int x,y,z;
        scanf("%s",ch);
        if(ch[0]=='A'){scanf("%d%d%d",&x,&y,&z);x++;y++;solveadd(x,y,z);}
        else {scanf("%d",&x);x++;printf("%lld\n",querysum(1,pos[x],pos[x]+s[x]-1));}
    }
    return 0;
}