ACM-ICPC 2018 沈阳赛区网络预赛 J. Ka Chang （分块+树状数组+dfs序）

题意
给你一颗树，由两种操作：
1.把这棵树深度为 D D <script type="math/tex" id="MathJax-Element-3">D</script>的点全部都加上一个值。
2.求以p为根节点的子树的权值和是多少？
思路
对于树上的东西，我们可以把他求一下DFS序，之后就可以把树上的结构变成线性的结构，之后就是查询和修改这两个东西了。
关于修改：对于修改来说，当前层的点少的时候我们直接树状数组暴力修改，对于当前层数的点太多的情况来说，我们把层数打上标记。
对于查询，我们先查询树状数组里面的值，之后在查询我们每一层中的标记，具体怎么查询，看代码把。
代码

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 1e5+10;

vector<int>V[maxn],D[maxn] , large;
int n,q , u,v , maxd;
int l[maxn] , r[maxn], de[maxn] , tar[maxn];
int cnt = 0;

struct BIT
{
    int tree[maxn];
    void init()
    {
        memset(tree,0,sizeof(tree));
    }
    void update(int pos,int val)
    {
        for(; pos <= n+1 ; pos += pos & -pos) tree[pos] += val;
    }
    long long query(int pos)
    {
        long long ans = 0;
        for(; pos > 0 ; pos -= pos & -pos) ans += tree[pos];
        return ans;
    }
}sum;
void dfs(int u,int d)
{
    l[u] = ++cnt;
    maxd = max(maxd,d);
    D[d].push_back(l[u]); // 我们把相同层数的点放进去 
    for(auto & i : V[u])
    {
        dfs(i,d+1);
    }
    r[u] = cnt; 
}

signed main()
{

    BIT sum;
    sum.init();
    maxd = 0;
    scanf("%d%d",&n,&q);
    cnt = 0 ;
    for(int i = 0 ; i < n-1 ; i++)
    {
        scanf("%lld%lld",&u,&v);
        V[u].push_back(v);  
    }
    dfs(1,0);
    int B = sqrt(n);
    for(int i = 0 ; i <= maxd ;i++)
    {
        if(D[i].size() > B) large.push_back(i); // 如果某一层中的点数大于我们的块数就放进去 
    }
    while(q--)
    {
        int op;scanf("%lld",&op);
        if(op == 1)
        {
            int d,va;scanf("%lld%lld",&d,&va);
            if(D[d].size() <= B)
            {
                for(auto &i : D[d]) {sum.update(i,va);}
            }
            else tar[d] += va;
        }
        else 
        {
            int p;scanf("%lld",&p);
            int L = l[p] , R = r[p];
            long long ans = sum.query(R) - sum.query(L-1);
            for(auto &i : large)
            {
                ans += (upper_bound(D[i].begin(),D[i].end(), R) - lower_bound(D[i].begin() , D[i].end() , L)) *tar[i]; // 关于查询，我们在每一层中查询看
                                                                                                                      // 看我们这一层中有多少个包含这个区间的值，然后我们乘上标记就好了 
            }
            cout<<ans<<endl;

        }
    }
}