【题解】CF1499F Diameter Cuts 树形dp、子树合并、 “DP 套 DP”

题目链接

一个自然的想法是枚举所有断边的方案，判断是否合法。

而对于求树的直径，可以这个树形 dp 求，那么我们考虑用树形 dp 的结果当作外层计数 dp 的状态，具体地：

• 令 \(f[u,h]\) 为考虑以 \(u\) 为根的子树，断边后 \(u\) 子树内距离 \(u\) 最远的点的距离为 \(h\) 的断边方案。

该 dp 转移类似树形背包的子树合并，考虑根节点到儿子的每条边是否断开。

复杂度为 \(O(n^2)\) 。

代码记录