一些有意思的题

一些有意思，但是没地方提交的题。

1

每个点有一个颜色，求树上所有颜色不同点对的距离和。

做法一

不同点对的距离和 (=) 总距离和 (-) 相同颜色的点对距离和。

对每个颜色建立虚树，然后算就得了。

做法二

首先同样进行第一步转化。

考虑使用线段树合并维护一个子树中每种颜色有多少个，深度之和为多少。

到一个结点处理两件事情：

• 不同子树之间的结点距离和，可以发现这些点的lca都是当前点，所以距离可算。
• 子树中的点到当前点的距离和。

可能这个方法会难写一些？