BZOJ 4034: [HAOI2015]T2

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

不想写树剖…所以考虑如何用dfs序解决，发现好像是可以的。入栈是1，出栈是-1，修改一个点就是入栈的时候加这个值，出栈的时间减去，修改子树就是从入栈到出栈，查询就是从1一直到入栈。因为是区间修改区间查询，所以就写线段树啦（dfs序+线段树，树剖即视感！）。flag维护区间里正值有多少个（可以为负),sum维护和,add维护懒标记。
Tips:除了开long long就没什么了，我一开始YY的dfs序做法是没有flag的，直接在l和r上加减，这样子树那边会出bug…

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=300000+10;
int l[maxn],r[maxn],dfs_clock,a[maxn],n,m,hm[maxn];
vector<int> g[maxn];
long long sum[maxn*4],add[maxn*4],flag[maxn*4];
void dfs(int p,int fa)
{
  l[p]=++dfs_clock;
  hm[dfs_clock]=1;
  for(int i=0;i<g[p].size();i++)
  {
    int v=g[p][i];if(v==fa) continue;
    dfs(v,p);
  }
  r[p]=++dfs_clock;
  hm[dfs_clock]=-1;
}
void pushdown(int p,int l,int r,int mid)
{
  if(add[p])
  {
    add[p*2]+=add[p];add[p*2+1]+=add[p];
    sum[p*2]+=(long long)flag[p*2]*add[p];sum[p*2+1]+=(long long)flag[p*2+1]*add[p];
    add[p]=0;
  }
}
void build(int p,int l,int r)
{
  if(l==r)
  {
    flag[p]=hm[l];
    return;
  }
  int mid=(l+r)>>1;
  build(p*2,l,mid);
  build(p*2+1,mid+1,r);
  flag[p]=flag[p*2]+flag[p*2+1];
}
void update(int p,int l,int r,int x,int y,long long v)
{
  if(l>y||r<x) return;
  if(l>=x&&r<=y)
  {
    add[p]+=v;
    sum[p]+=(long long)v*flag[p];
    return;
  }
  int mid=(l+r)>>1;
  pushdown(p,l,r,mid);
  update(p*2,l,mid,x,y,v);
  update(p*2+1,mid+1,r,x,y,v);
  sum[p]=sum[p*2]+sum[p*2+1];
}
long long query(int p,int l,int r,int x,int y)
{
  if(l>y||r<x) return 0;
  if(l>=x&&r<=y) return sum[p];
  int mid=(l+r)>>1;
  pushdown(p,l,r,mid);
  return query(p*2,l,mid,x,y)+query(p*2+1,mid+1,r,x,y);
}
int main()
{
  //freopen("data.in","r",stdin);
  //freopen("data.out","w",stdout);
  scanf("%d%d",&n,&m);
  for(int i=1;i<=n;i++) scanf("%d",&a[i]);
  for(int i=1;i<n;i++)
  {
    int x,y;
    scanf("%d%d",&x,&y);
    g[x].push_back(y);
    g[y].push_back(x);
  }
  dfs(1,-1);
  build(1,1,2*n);
  for(int i=1;i<=n;i++)
  {
    update(1,1,2*n,l[i],l[i],a[i]);
    update(1,1,2*n,r[i],r[i],a[i]);
  }
  int op,x,v;
  for(int i=1;i<=m;i++)
  {
    scanf("%d%d",&op,&x);
    if(op==1)
    {
      scanf("%d",&v);
      update(1,1,2*n,l[x],l[x],v);
      update(1,1,2*n,r[x],r[x],v);
    }
    if(op==2)
    {
      scanf("%d",&v);
      update(1,1,2*n,l[x],r[x],v);
    }
    if(op==3) printf("%lld\n",query(1,1,2*n,1,l[x]));
  }
  return 0;
}