题目大意 给定一颗树，然后每个结点有一种颜色，然后求所有路径中不同颜色的个数之和。 思路 可以这样想，先假设所有路径中都包含所有的n种颜色，则可以得出ans = n * n * (n - 1) / 2 然后，减去路径中不存在某个颜色的路径条数，可以想象一下一棵树中某个联通分量中没有某个颜色，然后这个联通分量点的个数为cnt那么就需要在ans中减cnt * （cnt - 1） / 2 代码