前置题目:树的颜色(时序差分)。
每种颜色的贡献为至少出现一次的路径数。容斥,总路径数减去未出现过的路径数。
考虑颜色 c c c 的贡献,此时将所有颜色为 c c c 的点移去,会形成若干连通块。对于大小为 x x x 的连通块,其贡献为 x ( x − 1 ) 2 \frac{x(x-1)}{2} 2x(x−1)。
在这些颜色为 c c c 的点(以及根节点)处计算对应块的贡献即可。
记 f c ( u ) f_c(u) fc(u) 表示以 u u u 为根子树内,无颜色
前置题目:树的颜色(时序差分)。
每种颜色的贡献为至少出现一次的路径数。容斥,总路径数减去未出现过的路径数。
考虑颜色 c c c 的贡献,此时将所有颜色为 c c c 的点移去,会形成若干连通块。对于大小为 x x x 的连通块,其贡献为 x ( x − 1 ) 2 \frac{x(x-1)}{2} 2x(x−1)。
在这些颜色为 c c c 的点(以及根节点)处计算对应块的贡献即可。
记 f c ( u ) f_c(u) fc(u) 表示以 u u u 为根子树内,无颜色