题解 P7288 「EZEC-5」树木的愤怒

Description

删除两条边，将树分成三个连通块 $A,B,C$，贡献为连通块点权和之积

删去一条边，问再删去任意一条边后的贡献和

Solution

考虑预处理出所有答案

先将边下放到子节点

$O(n^2)$ 做法：

枚举删除的任意两条边，算出贡献，累加到两条边上

考虑优化

发现删去一条边后，再删第二条边时，对于很多情况，都会有一个连通块不变，考虑剩下连通块的贡献一起算，再乘上这个连通块的贡献

分类讨论

定义 $s_u$ 为子树 $u$ 的点权和

• 删的第二条边在子树内

贡献为

$$\begin{gathered} (s_1-s_u)\sum _{v\in so{n}_u}(s_u-s_v)s_v \\ =(s_1-s_u)\sum _{v\in so{n}_u}(s_u{s}_v-s_v^2) \\ =(s_1-s_u)(s_u\sum _{v\in so{n}_u}{s}_v-\sum _{v\in so{n}_u}{s}_v^2)) \end{gathered}$$

将 $s$ 做子树和，平方的子树和，$O(n)$ 统计贡献

• 删的第二条边在到根的路径上

贡献为

s u ∑ v ∈ r o a d 1 , u ( s 1 − s v ) ( s v − s u ) = s u ∑ v ∈ r o a d 1 , u ( s 1 s v − s 1 s u − s v 2 + s u s v ) = s u ( s 1 ∑ v ∈ r o a d 1 , u s v − ∑ v ∈ r o a d 1 , u s 1 s u − ∑ v ∈ r o a d 1 , u s v 2 + s u ∑ v ∈