一些有意思,但是没地方提交的题。
每个点有一个颜色,求树上所有颜色不同点对的距离和。
不同点对的距离和 (=) 总距离和 (-) 相同颜色的点对距离和。
对每个颜色建立虚树,然后算就得了。
首先同样进行第一步转化。
考虑使用线段树合并维护一个子树中每种颜色有多少个,深度之和为多少。
到一个结点处理两件事情:
lca
可能这个方法会难写一些?