codeforces 396C On Changing Tree dfs序+BIT

    一棵树,根节点为1,现有两种操作

1 v x k :对节点v的值+x,对节点v的后继中,距离为i的节点的值+x-i*k。

2 v :输出节点v现在的值。

对子树进行操作的话,容易想到树状数组或者线段树,先对树做一遍dfs标上时间戳,这样节点v的子树的时间戳就是l[v]--r[v]这个区间,那么对子树的加减就可以转化成区间的加减。再来看操作,操作一v的操作可以看成+x-0*k,或者说是x+deep[v]*k-deep[v]*k,对于v的一个孩子u,就是x+deep[v]*k-deep[u]*k,对于一个距离v为i的后继t,就是x+deep[v]-deep[t]*k,可以找到这几项加的项是相同的,而减的项取决于当前的节点,那么每次操作1的时候,就可以把加的项和减的项用梁颗线段树或者两个树状数组分别统计起来,减的项不同层的值虽然不同,但是k是相同的,而deep可以在dfs的时候存好,查询的时候直接用,所以减的项存k的总和就行,区间查找+单点查询,线段树和树状数组都可以实现,树状数组还好写点...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn=330000;
const ll MOD=1000000007;

int n,m;

struct BIT
{
    ll dt[maxn];
    void init()
    {
        memset(dt,0,sizeof dt);
    }
    int lowbit(int x)
    {
        return (-x)&x;
    }
    void modify(int x,ll c)
    {
        if (x==0 || x>n) return;
        for (int x1=x; x1<=n; x1+=lowbit(x1))
        {
            dt[x1]+=c;
            dt[x1]%=MOD;
        }
    }
    ll query(int x)
    {
        ll res=0;
        for (int x1=x; x1>0; x1-=lowbit(x1))
        {
            res+=dt[x1];
            res%=MOD;
        }
        return res;
    }
}bit[2];

vector<int> g[maxn];
int l[maxn],r[maxn];
int tt;
ll dis[maxn];
void dfs(int p,int fa)
{
    if (p>1) dis[p]=dis[fa]+1;
    tt++;
    l[p]=tt;
    int v;
    for (int i=0; i<g[p].size(); i++)
    {
        v=g[p][i];
        dfs(v,p);
    }
    r[p]=tt;
}

int main()
{
//    freopen("in.txt","r",stdin);
    scanf("%d",&n);

    memset(l,0,sizeof l);
    memset(r,0,sizeof r);
    int u,v;
    for (int i=1; i<=n; i++) g[i].clear();
    for (int i=2; i<=n; i++)
    {
        scanf("%d",&v);
        g[v].push_back(i);
    }
    tt=0;
    memset(dis,0,sizeof dis);
    dfs(1,-1);
//    for (int i=1; i<=n; i++)
//    cout<<l[i]<<" "<<r[i]<<endl;
//        cout<<dis[i]<<endl;
    bit[0].init();
    bit[1].init();
    scanf("%d",&m);
    int tp;
    ll x,y,z;
    while(m--)
    {
        scanf("%d",&tp);
        if (tp==1)
        {
            scanf("%I64d%I64d%I64d",&x,&y,&z);
            bit[0].modify(l[x],y+dis[x]*z);
            bit[0].modify(r[x]+1,-y-dis[x]*z);
            bit[1].modify(l[x],z);
            bit[1].modify(r[x]+1,-z);
        }
        else
        {

            scanf("%d",&v);
            ll ans=bit[0].query(l[v])-(bit[1].query(l[v])*dis[v])%MOD;
            ans=(ans%MOD+MOD)%MOD;
            cout<<ans<<endl;
        }
    }
    return 0;
}


你可能感兴趣的:(codeforces 396C On Changing Tree dfs序+BIT)