luogu P2664 树上游戏 题解

$\mathtt{l}\mathtt{u}\mathtt{o}\mathtt{g}\mathtt{u}\mathtt{P}2664树上游戏$

题目大意：

一棵树上每个节点都有颜色，给定一个长度为 $n$ 的颜色序列，定义 $s(i,j)$ 为 $i\to j$ 的颜色数量。

设 $su{m}_i={\sum }_{j=1}^{n}s(i,j)$。求所有的 $su{m}_i$。

---

这个本质可以看成树上的点对的问题。既然不考虑在线我们可以使用离线算法点分治。

我们需要在 $O(n)$ 或者 $O(n\log n)$ 的复杂度完成统计以当前节点为 $lca$ 的点对的所有贡献。

显然暴力统计是不行的。考虑借鉴一下统计树上距离为 $k$ 的点对的方法，开桶进行计算。

首先考虑以根节点为起点的方案数，也就是考虑每一条链上第一个出现的节点，其对于根的贡献是其子树大小。

其子树中每一个点都会产生一次这样颜色的贡献。

为了方便统计我们将根节点的颜色也加上贡献，同时提前打上标记。

之后需要统计点对的贡献。我们不妨将上一次产生贡献的点的贡献记成 $f(i)$。那么不产生贡献显然是 $f(i)=0$。

既然是统计子树中的贡献。考虑遍历每一个子树。

对于每个点的贡献就是。设当前的颜色集合为 $S$。点集合是 $T$，子树的大小是 $Siz$。

∑ f − ∑ v ∈ T f ( v )