bzoj4543: [POI2014]Hotel加强版（长链剖分+dp）

传送门
代码：
长链剖分好题。
题意：给你一棵树，问树上选三个互不相同的节点，使得这个三个点两两之间距离相等的方案数。

---

思路：
先考虑$dp$。
$f_{i,j}$表示$i$子树中离$i$距离为$j$的点数，$g_{i,j}$表示$i$子树中所有满足$dist(lca(u,v),i)-dist(lca(u,v),i)=j$的点对数。
不难发现这两个状态是可以进行拼接的。
因此对于一对父亲和孩子$(p,v)$有如下转移式：
a n s + = g v , i + 1 ∗ f