题目大意:给一棵树,树上的点初始化是0,现在给出两种操作
1 v x k:v这个结点加上x,然后和v距离为i的所有子结点加上 x - (i * k)
2 v:询问v这个节点的值
解题思路:先dfs处理出每个节点所包含的子孙结点的区域l和r,还有每个点的深度dep,
假设我Modify(l[v], x + k * dep[v])和Modify(r[v] + 1, -(x + k * dep[v])),然后用一个树状数组维护前缀和,这个树状数组设为bit1
观察修改可得,v这个数多加上了k * dep[v]的值,所以要减去,而后面的所有的v的子结点也要减去相应的值
这里可以用深度入手,因为深度每增加一,意味着就要减少k,也就是说(j是v的子结点),要减少的数是dep[j] * k,这是因为bit1树状数组维护的是前缀和,所以要减去,就相当于x + k * dep[v] - k * dep[j],化简可得x + k * (dep[v] - dep[j]),而dep[j] - dep[v]刚好是亮点之间的距离,所以这刚好是所需要的数
所以我们再开另外一个树状数组维护前缀k, 假设为bit2,bit2维护的也是前缀和
那么所得到的结果就是 Query(l[v], bit1) - Query(l[v], bit2) * dep[v]了
#include <cstdio>
#include <cstring>