[IOI2009]Regions

Regions

题解

题目相当于让我们求颜色为$r1$的点$u$与颜色为$r2$的点$v$之间是祖先儿子关系的这样的$(u,v)$的对数。
我们可以先考虑将数据离线下来，如果每次只处理两个颜色之间的就太浪费时间了，题目的$R$明明比$Q$小这么多。
我们可以进行一次$dfs$，到达每个节点时，记录下它的祖先链上的每种颜色的点的个数，对于询问$(r1,r2)$，我们就在枚举到颜色为$r2$的节点时将它祖先中颜色为$r1$的节点的数量加入答案。

但很明显，如果我们大部分询问都在问一种颜色的节点，而这个颜色的节点数量又比较多，那么这种操作就会执行很多次，然后，就$T$了。
所以我们可以考虑按查询的次数进行分块，对于颜色$i$，如果它在询问中从当$r2$的次数不超过$S$次，那么我们就使用上面的方法统计答案。
这样下来，我们在$dfs$遍历树时每个节点最多会进行$S$次操作，这个部分时间复杂度是$O\left(nS\right)$的。

而对于出现次数大于$S$的颜色，它们最多会有$\frac{q}{S}$个，我们可以对这些颜色都进行一次$dfs$。 d f s