题意:
给出一棵 n 个点的无根树,请在这棵树上选三个互不相同的节点,使得这个三个
节点两两之间距离相等,输出方案数即可。
分析:
定义
设 f ( x , i ) f(x,i) f(x,i)表示在以x为根的子树中,与x距离为i的节点数
g ( x , i ) g(x,i) g(x,i)表示在以x为根的子树中选择了两个节点,最后一个点需满足与x的距离为i的方案数。
所以 f ( x , i ) = ∑ u f ( u , i − 1 ) f(x,i)=\sum_u f(u,i-1) f(x,i)=∑uf(u,i−1)
g ( x , i ) = ( ∑ u g ( u , i + 1 ) ) + ( ∑ u , v f ( u , i ) ∗ f ( v , i ) ) g(x,i)=(\sum_u g(u,i+1))+(\sum_{u,v}f(u,i)*f(v,i)) g(x,i)=(∑ug(u,i+1))+(∑u,vf<