【FZU 2277】Change

题意

   有一颗有n个节点的有根树，根节点编号时1，每个结点都有一个值ai，开始的时候，所有节点的值都是0.

   我们有q个操作，操作只有两种类型

   1 v x k，a[v]+=x,a[v']+=x-k,a[v"]+=x-2*k... v'是结点v的孩子 。

  2 v 输出a[v]mod 1e9+7。

分析

dfs序+线段树

下面的代码TLE了，一会再改，感觉没啥毛病哇

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
typedef long long LL;

const int maxn=300000+100;
const int MOD=1000000007;
int n,T,q,num ;
vector<int>child[maxn];
LL a[maxn],r[maxn],dep[maxn],val[maxn];
void dfs(LL u){
    num++;
    a[num]=u;
    for(int i=0;i<child[u].size();i++){
        int v=child[u][i];
        dep[v]=dep[u]+1;
        dfs(v);
    }
    r[u]=a[num];
    return ;
}
LL addv[4*maxn],kv[4*maxn];
int ql,qr;
LL v,k;

void update(int o,int L,int R){
    if(ql<=L&&qr>=R){
        addv[o]=(add[o]%MOD+v%MOD)%MOD;
        kv[o]=(kv[o]%MOD+k%MOD)MOD;
        return;
    }
    int M=L+(R-L)/2;
    if(ql<=M)
       update(2*o,L,M);
    if(qr>M)
       update(2*o+1,M+1,R);
    return;
}
//找到v的值
LL query(int o,int L,int R,LL adv,LL adk){
    if(L==R){
        return addv[o]+adv-dep[val[L]]*(kv[o]+adk);
    }
    int M=L+(R-L)/2;
    if(v<=M)
        return query(2*o,L,M,(adv%MOD+addv[o]%MOD)%MOD,(adk+kv[o])%MOD);
    if(v>M)
        return query(2*o+1,M+1,R,(adv%MOD+addv[o]%MOD)%MOD,(adk%MOD+kv[o]%MOD)%MOD);
}
int main(){
    scanf("%d",&T);
    for(int t=1;t<=T;t++){
        scanf("%d",&n);
        for(int i=1;i<=n;i++)child[i].clear();
        num=0;
        for(int i=2;i<=n;i++){
            int x;
            scanf("%d",&x);
            child[x].push_back(i);
        }
        dep[1]=1;
        dfs(1);
        for(int i=1;i<=n;i++)val[a[i]]=i;
        memset(addv,0,sizeof(addv));
        memset(kv,0,sizeof(kv));
        scanf("%d",&q);
        for(int i=1;i<=q;i++){
            int ty;
            scanf("%d",&ty);
            if(ty==1){
               int vv;
               LL x;
               scanf("%d%lld%lld",&vv,&x,&k);
               ql=a[vv],qr=r[vv],v=x+dep[vv]*k;
              // cout<<ql<<" "<<qr<<endl;
               update(1,1,n);
            }else{
                int vv;
                scanf("%d",&vv);
                v=a[vv];
                cout<<query(1,1,n,0,0)%MOD<<endl;
            }
        }

       /* for(int i=1;i<=num;i++)
            printf("%d ",a[i]);
        printf("
");
        for(int i=1;i<=n;i++){
            printf("%d %d
",i,r[i]);
        }*/
    }
return 0;
}