51nod 1199 Money out of Thin Air（线段树+树剖分）

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1199

题意：

思路：
因为是一棵树，所以需要把它剖分一下再映射到线段树上，剖分的话只需要dfs一遍树即可，得到的dfs序就是每个结点在线段树中的位置，子树上的节点的编号都是连续的。

接下来的操作就是线段树的查询和更新了，这部分并不难。

  1 #include<iostream>
  2 #include<algorithm>
  3 #include<cstring>
  4 #include<cstdio>
  5 #include<vector>
  6 #include<stack>
  7 #include<queue>
  8 #include<cmath>
  9 #include<map>
 10 #include<set>
 11 using namespace std;
 12 typedef long long ll;
 13 const int INF = 0x3f3f3f3f;
 14 const int maxn=50000+5;
 15
 16 ll n, m, dfs_clock;
 17 int w[maxn],son[maxn],dfn[maxn],num[maxn];
 18 ll lazy[maxn<<2];
 19
 20 vector<int> G[maxn];
 21
 22 struct node
 23 {
 24     int l, r;
 25     ll sum;
 26 }t[maxn<<2];
 27
 28 void dfs(int u)
 29 {
 30     dfn[u]=++dfs_clock;
 31     son[u]=1;
 32     for(int i=0;i<G[u].size();i++)
 33     {
 34         int v=G[u][i];
 35         dfs(v);
 36         son[u]+=son[v];
 37     }
 38 }
 39
 40 void PushUp(int o)
 41 {
 42     t[o].sum=t[o<<1].sum+t[o<<1|1].sum;
 43 }
 44
 45 void PushDown(int o)
 46 {
 47     if(lazy[o])
 48     {
 49         lazy[o<<1]+=lazy[o];
 50         lazy[o<<1|1]+=lazy[o];
 51         int m=t[o].r-t[o].l+1;
 52         t[o<<1].sum+=lazy[o]*(m-(m>>1));
 53         t[o<<1|1].sum+=lazy[o]*(m>>1);
 54         lazy[o]=0;
 55     }
 56 }
 57
 58 void build(int l ,int r, int o)
 59 {
 60     lazy[o]=0;
 61     t[o].l=l;
 62     t[o].r=r;
 63     t[o].sum=0;
 64     if(l==r)
 65     {
 66         t[o].sum=num[l];
 67         return;
 68     }
 69     int mid=(l+r)>>1;
 70     build(l,mid,o<<1);
 71     build(mid+1,r,o<<1|1);
 72     PushUp(o);
 73 }
 74
 75 void update(int ql, int qr, int l, int r, ll z, int o)
 76 {
 77     if(ql<=l && qr>=r)
 78     {
 79         t[o].sum+=(r-l+1)*z;
 80         lazy[o]+=z;
 81         return;
 82     }
 83     PushDown(o);
 84     int mid=(l+r)>>1;
 85     if(ql<=mid)  update(ql,qr,l,mid,z,o<<1);
 86     if(qr>mid)  update(ql,qr,mid+1,r,z,o<<1|1);
 87     PushUp(o);
 88 }
 89
 90 ll query(int ql, int qr, int l, int r, int o)
 91 {
 92     if(ql<=l && qr>=r)  return t[o].sum;
 93     PushDown(o);
 94     int mid=(l+r)>>1;
 95     ll ans=0;
 96     if(ql<=mid)  ans+=query(ql,qr,l,mid,o<<1);
 97     if(qr>mid)   ans+=query(ql,qr,mid+1,r,o<<1|1);
 98     return ans;
 99 }
100
101 int main()
102 {
103     //freopen("in.txt","r",stdin);
104     scanf("%d%d",&n,&m);
105     for(int i=0;i<n;i++)  G[i].clear();
106     for(int i=1;i<n;i++)
107     {
108         int p; scanf("%d%d",&p,&w[i]);
109         G[p].push_back(i);
110     }
111     w[0]=0;
112     dfs_clock=0;
113     dfs(0);
114     for(int i=0;i<n;i++)  num[dfn[i]]=w[i];
115     build(1,n,1);
116     char op[3]; ll x, y, z;
117     memset(lazy,0,sizeof(lazy));
118     while(m--)
119     {
120         scanf("%s%lld%lld%lld",op,&x,&y,&z);
121         if(op[0]==‘S‘)
122         {
123             if(query(dfn[x],dfn[x],1,n,1)<y)  update(dfn[x],dfn[x],1,n,z,1);
124         }
125         else
126         {
127             if(query(dfn[x],dfn[x]+son[x]-1,1,n,1)<son[x]*y)  update(dfn[x],dfn[x]+son[x]-1,1,n,z,1);
128         }
129     }
130     for(int i=0;i<n;i++)  printf("%lld\n",query(dfn[i],dfn[i],1,n,1));
131     return 0;
132 }