点分治/dsu on tree可以 $\Theta(n\log n)$ 求出来 $a_i$ 表示树上长度为 $i$ 的链的数量吗？$i$ 取遍 $1\sim n$。