F - Change FZU - 2277 （DFS序+线段树）

题目链接：

F - Change

FZU - 2277

题目大意：

题意: 给定一棵根为1, n个结点的树. 有q个操作,有两种不同的操作

(1) 1 v k x : a[v] += x, a[v '] += x – k(v '为v的儿子), a[v ' '] += x – 2 * k(v ' '是v '的儿子) ... ;

(2) 2 v : 输出a[v] % (1e9 + 7);

具体思路：dfs序+线段树

a[v'']=a[v'']+x-k*(depth[v'']-depth[v])=a[v'']+x-k*depth[v"]+k*depth[v]=(a[v"]+k*depth[v]) - k*depth[v""].

第一个括号里面用一个线段树维护，第二个括号里面用另一个线段树维护就可以了。

具体求值的时候：对于一次增加操作，节点x增加的值其实就是x-(deep[x]-deep[v])*k。（deep[v]为该次修改的根节点）。

AC代码：

 #include<iostream>
 #include<stdio.h>
 #include<vector>
 #include<cstring>
 using namespace std;
 # define ll long long
 # define lson l,mid,rt<<
 # define rson mid+,r,rt<<|
 const int maxn = 3e5+;
 const int mod  =1e9+;
 ll tree1[maxn<<],tree2[maxn<<];
 int dfsord,n;
 vector<int>edge[maxn];
 int st[maxn],ed[maxn],depth[maxn];
 int Scan()
 {
     int res = , ch, flag = ;
     if((ch = getchar()) == '-')
         flag = ;
     else if(ch >= '' && ch <= '')
         res = ch - '';
     while((ch = getchar()) >= '' && ch <= '' )
         res = res *  + ch - '';
     return flag ? -res : res;
 }

 ll Scan_l()
 {
     ll res = ;
     int ch, flag = ;
     if((ch = getchar()) == '-')
         flag = ;
     else if(ch >= '' && ch <= '')
         res = ch - '';
     while((ch = getchar()) >= '' && ch <= '' )
         res = res *  + ch - '';
     return flag ? -res : res;
 }
 void init()
 {
     for(int i=; i<n; i++)
     {
         edge[i].clear();
     }
     dfsord=;
     memset(tree1,,sizeof(tree1));
     memset(tree2,,sizeof(tree2));
 }
 void dfs(int cur,int fa,int dep)
 {
     st[cur]=ed[cur]=++dfsord;
     depth[dfsord]=dep+;
     for(int i=; i<edge[cur].size(); i++)
     {
         int to=edge[cur][i];
         if(to==fa)
             continue;
         dfs(to,cur,dep+);
     }
     ed[cur]=dfsord;
 }
 void down(int rt)
 {
     tree1[rt<<]=(tree1[rt<<]+tree1[rt]+mod)%mod;
     tree1[rt<<|]=(tree1[rt<<|]+tree1[rt]+mod)%mod;
     tree2[rt<<]=(tree2[rt<<]+tree2[rt]+mod)%mod;
     tree2[rt<<|]=(tree2[rt<<|]+tree2[rt]+mod)%mod;
     tree1[rt]=,tree2[rt]=;
 }
 void update(int l,int r,int rt,int L,int R,ll val1,ll val2)
 {
     if(L<=l&&R>=r)
     {
         tree1[rt]=(tree1[rt]+val1+mod)%mod;
         tree2[rt]=(tree2[rt]+val2+mod)%mod;
         return ;
     }
     down(rt);
     int mid=(l+r)>>;
     if(L<=mid)
         update(lson,L,R,val1,val2);
     if(R>mid)
         update(rson,L,R,val1,val2);
 }
 ll ask(int l,int r,int rt,int pos)
 {
     if(l==r)
     {
         return (tree1[rt]+tree2[rt]*depth[pos]%mod+mod)%mod;
     }
     int mid=(l+r)>>;
     down(rt);
     if(pos<=mid)
         return ask(lson,pos);
     else
         return ask(rson,pos);
 }
 int main()
 {
     int T,tmp;
     T=Scan();
     while(T--)
     {
         n=Scan();
         init();
         for(int i=; i<=n; i++)
         {
             tmp=Scan();
             edge[tmp].push_back(i);
         }
         int m;
         dfs(,,);
         m=Scan();
         while(m--)
         {
             int op,v;
             ll x,k;
             op=Scan();
             if(op==)
             {
                 v=Scan();
                 x=Scan_l();
                 k=Scan_l();
                 update(,n,,st[v],ed[v],x+k*depth[st[v]],-k);
             }
             else
             {
                 v=Scan();
                 ll ans=ask(,n,,st[v]);
                 ans=(ans%mod+mod)%mod;
                 printf("%lld\n",ans);
             }
         }
     }
     return ;
 }