Gym - 100589A Queries on the Tree 树状数组+分块

题目链接：https://vjudge.net/problem/Gym-100589A

题意：n个点，根节点为1的树，两种操作，1 L y 与根节点距离为L的节点权值全部加上y，2 x x子树的权值总和

题解：对于更新操作，因为更新的是某一个深度，若对该深度的每个点都更新，那么对于每次更新就是O(n)的复杂度，查询就是logn，如果只对每一个更新的深度进行标记，那么查询的时候就查询这个字数的dfs序在每一个深度出现的数目是多少，这样二分就可以实现，这样跟新就是O(1)，查询就是nlogn，显然都是不可行的，因此我们有个让更新和查询都将就一点的方法就是分块，我们把每一层的点数按照sqrt(n)的数目进行划分，小于sqrt(n)的直接更新，大于的就标记层数，这样查询的时候也最多查询n/sqrt(n)层，所以这样复杂度就是mlog(n)sqrt(n)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define lowbit(x) (x & (-x))
const int N = 1e5 + 10;
ll sum[N];
int n, m, limit;
vector<int> v[N], dep[N], maxdep;
ll val[N];
int in[N], out[N], tot;
void dfs(int u, int fa, int deep) {
	in[u] = ++tot;
	dep[deep].push_back(in[u]);
	int to;
	for(int i = 0; i < v[u].size(); i++) {
		to = v[u][i];
		if(to == fa) continue;
		dfs(to, u, deep + 1);
	}
	out[u] = tot;
}
void update(int pos, int val) {
	while(pos <= n) {
		sum[pos] += val;
		pos += lowbit(pos);
	}
}
ll query(int pos) {
	ll res = 0;
	while(pos) {
		res += sum[pos];
		pos -= lowbit(pos);
	}
	return res;
}
int main() {
	int op;
	int x, y, L;
	scanf("%d %d", &n, &m);
	limit = sqrt(n);
	for(int i = 1; i <= n - 1; i++) {
		scanf("%d %d", &x, &y);
		v[x].push_back(y);
	}
	dfs(1, 0, 0);
	for(int i = 0; i < n; i++) {
		if(dep[i].size() >= limit) {
			maxdep.push_back(i);
		}
			
	}
	ll ans;
	while(m--) {
		scanf("%d", &op);
		if(op == 1) {
			scanf("%d %d", &L, &y);
			if(dep[L].size() < limit) {
				for(int i = 0; i < dep[L].size(); i++)
					update(dep[L][i], y);
			} else {
				val[L] += y;
			}
		} else {
			scanf("%d", &x);
			ans = query(out[x]) - query(in[x] - 1);
			for(int i = 0; i < maxdep.size(); i++) {
				ans += val[maxdep[i]] * (upper_bound(dep[maxdep[i]].begin(), dep[maxdep[i]].end(), out[x]) - lower_bound(dep[maxdep[i]].begin(), dep[maxdep[i]].end(), in[x]));	
			}
				
			printf("%lld\n", ans);
		}
	}
	return 0;
}