小H和游戏——dfs序+树状数组 | 思维

题目链接：https://ac.nowcoder.com/acm/contest/5203/D

题目大意：

给你一棵n个节点的树，然后有q次轰炸，每次轰炸可以波及到与当前节点距离小于等于2的节点，并输出该次轰炸后当前节点的被轰炸次数。

题解：

解法一：树状数组（线段树）+dfs序

这个题先讲相对而言时间复杂度比较高的解法，但是比较好想，容易理解。

首先我们可以求出每个点的儿子编号的区间（类似于dfs序的操作）以及每个节点的编号。

求出区间后就简单了，考虑每次轰炸，只把贡献传给儿子和父亲，这样统计的时候查询当前节点的值（即轰炸儿子和父亲产生的贡献），父亲节点的值（即轰炸爷爷节点和兄弟节点产生的贡献）以及儿子区间的值（轰炸孙子节点产生的贡献）

经典的区间修改，区间查询操作，推荐用树状数组维护（常数小）

trick:有一部分多考虑了，需要减去，代码中有解释。

#pragma GCC optimize(2)
#include
#include
#include
#include
#include
#include
#include
#include

解法二：思维

首先我们不难发现对一个节点产生贡献的节点有6种，自身，儿子，孙子，兄弟，父亲，爷爷。

父亲节点，爷爷节点比较好办，因为就一个父亲和爷爷。

那儿子和孙子节点怎么处理呢？先不考虑最难的兄弟节点

那就不处理呗，每次只是暴力地把贡献上传给父亲和爷爷。

这样对于父亲和爷爷来说，他们儿子和孙子的节点贡献事先已经求出来了。

然后考虑最难的兄弟节点的贡献。

我们设cnt[i][0/1/2]表示一次轰炸后，自己/儿子/孙子波及到i点且还剩距离j可以波及的轰炸的次数

兄弟的贡献，一定是保存在了cnt[fa[i]][1]当中的，不过，里面有一个是自己波及的，所以答案加上cnt[fa[i]][1]-cnt[x][2]即可

代码实现：

#pragma GCC optimize(2)
#include
#include
#include
#include
#include
#include
#include
#include