bzoj4034 [HAOI2015]T2

　　首先求出树的dfs序，对于第一种操作和第二种操作分开处理，第一种操作比较经典，就不说了。第二种操作其实和第一种也很类似，查询x点到根的权值和sum(x)=(disx+1)*v(x)-w，其中disx为x点到根的距离，v(x)表示以x到根之间的点为第二类操作的子树根所进行操作的权值和，这样相乘明显会有多余的部分，假设对y点进行第二类操作,使其和其子树上的点增加z，那么会多计算价值disy*z，使用两颗树状数组维护即可。

　　代码

　　

#include<cstdio>
#define nc getchar
#define N 1000010
int tot,n,m,i,v[N],dp,p[N],pre[N],tt[N],L[N],R[N],a,b,T;
long long c[N][3],dis[N],ans;
inline int lowbit(int x)
{
    return x&(-x);
}
void cc(int x,long long w,int typ)
{
    while (x<=tot)
    {
        c[x][typ]+=w;
        x+=lowbit(x);
    }
}
long long sum(int x,int typ)
{
    long long ans=0;
    while (x)
    {
        ans+=c[x][typ];
        x-=lowbit(x);
    }
    return ans; 
}
void link(int x,int y)
{
    dp++;pre[dp]=p[x];p[x]=dp;tt[dp]=y;
}
void dfs(int x,int fa)
{
    int i=p[x];
    tot++;L[x]=tot;
    while (i)
    {
        if (tt[i]!=fa)
        {
            dis[tt[i]]=dis[x]+1;
            dfs(tt[i],x);
        }
        i=pre[i];
    }
    tot++;R[x]=tot;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++)
    scanf("%d",&v[i]);
    for (i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        link(a,b);link(b,a);
    }
    dfs(1,0);
    for (i=1;i<=n;i++)
    {
        cc(L[i],v[i],0);
        cc(R[i],-v[i],0);
    }
    for (i=1;i<=m;i++)
    {
        scanf("%d",&T);
        if (T==1)
        {
            scanf("%d%d",&a,&b);
            cc(L[a],b,0);
            cc(R[a],-b,0);
        }
        else
        if (T==2)
        {
            scanf("%d%d",&a,&b);
            cc(L[a],b,1);
            cc(R[a],-b,1);
            cc(L[a],dis[a]*(long long)b,2);
            cc(R[a],-dis[a]*(long long)b,2);
        }
        else
        {
            scanf("%d",&a);
            ans=sum(L[a],0)-sum(L[a],2)+sum(L[a],1)*(dis[a]+1);
            printf("%lld
",ans);
        }
    }
    
}