蒟蒻自己想的。

先找出树的直径，类似换根 dp 求出经过直径的路径的贡献，然后把直径上所有点都删去，剩下的连通块递归处理。

不知道这种能不能卡掉。