华华和月月种树（dfs序+树状数组）

链接：https://ac.nowcoder.com/acm/problem/23051
来源：牛客网

题意

对一棵树（初始只有 0 号节点，权值为 0）进行操作和询问：

操作 1：输入格式1 i，表示使节点 i 长出了一个新的儿子节点，权值为0，编号为当前最大编号 +1（也可以理解为，当前是第几个操作 1，新节点的编号就是多少）。
操作 2：输入格式 2 i a，表示华华上线做任务使节点 i 的子树中所有节点（即它和它的所有子孙节点）权值加 a 。

询问 3：输入格式3 i，华华需要给出 i 节点此时的权值。

思路

先将树变成dfs序，然后就是树状数组了。当在节点 i 进行操作 2 的时候，就相当 [ dfn[i] , dfn[i]+sz[i]-1 ]范围加上a。当进行操作 1 的时候，只需要把新加入的节点权值变成0即可。

 

#include <bits/stdc++.h>
using namespace std;

const int maxn=1e5+10;
vector<int>v[maxn];
int n,m,cnt;
int dfn[maxn];//dfs序，相当于节点在树状数组中的下标
int sz[maxn]; //sz[i] 表示节点 i 为根的子树的所有节点数z
int tree[maxn];
struct query
{
    int op,i,a;
}q[maxn<<2];

int sum(int x)
{
    int s=0;
    while (x>0)
    {
        s+=tree[x];
        x-=(x&-x);
    }
    return s;
}

void update(int x,int s)
{
    while(x<=cnt)
    {
        tree[x]+=s;
        x+=(x&-x);
    }
}

void dfs(int u)
{
    sz[u]++;
    dfn[u]=++cnt;
    for(int i:v[u])
    {
        dfs(i);
        sz[u]+=sz[i];
    }
}

void solve() {
    scanf("%d",&m);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&q[i].op,&q[i].i);
        if(q[i].op==1)
        {
            v[q[i].i].push_back(++n);
            q[i].i=n; //记录新加入的子节点
        }
        else if(q[i].op==2)
            scanf("%d",&q[i].a);
    }
    dfs(0);
    for(int i=0;i<m;i++)
    {
        if(q[i].op==1)
        {
            int v=sum(dfn[q[i].i]);
            update(dfn[q[i].i],-v);
            update(dfn[q[i].i]+1,v);
        }
        else if(q[i].op==2)
        {
            update(dfn[q[i].i],q[i].a);
            update(dfn[q[i].i]+sz[q[i].i],-q[i].a);
        }
        else 
            printf("%d\n",sum(dfn[q[i].i]));
    }
}

signed main() {
    solve();
    return 0;
}