[GXOI/GZOI2019]旧词

题目

传送门 to luogu

思路

直接做是很困难的。

考虑这样构造一个可以做的实值函数 $v(x)$ ，满足 $$\sum _{u\in Path(x,root)}v(u)=depth(x{)}^k$$

也就是说，每个祖先（包括自己）的 $v$ 之和，恰好为 $depth(x{)}^k$ 。

如果我们成功的构造了这样的函数，显然，$depth(lca{)}^k$ 就被 分摊到了 $lca$ 到根的路径上。

此时，我们只需要求 $$\sum _{u\in Path(Y,root)}v(u)cnt(u)$$

其中 $cnt(u)$ 表示以 $u$ 为根的子树中，有多少个点编号在 $[1,X]$ 范围内（此 $X$ 为询问中的 $X$ ）。

其实这是很显然的，不妨看图：

此时，$lca(y,x_1)$ 的值已经被分摊到了其到根的路径上——因为这个点的每个祖先的 $cnt$ 都增大了 $1$ （并且被 $y$ 所计算到了），所以总贡献就是其祖先（包括自己）的 $v$ 之和，也即其 $dept{h}^k$ 。$x_2$ 同理。

那么这个 $v$ 应该是多少呢？其实很简单，构造一个即可：$$v(x)=depth(x{)}^k-{\left[depth(x)-1\right]}^k$$

正确性是显然的。这里我们充分运用了 $depth(x)-1=depth(f{a}_x)$ 的特点。

然后我们就可以做了！只需要 树链剖分（或者 $LCT$ ）即可。

按照 $X$ 排序，每次插入，只是维护一个 $cnt$ （和 $cnt\times v$ ）而已。毕竟 $v$ （及 $v$ 的和）可以预处理。

代码

咕咕咕……
我真的不想打了……