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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值