题目链接:HDU - 6035 - Colorful Tree(树形dp)
先贴一个巨佬博客 hdu_6032:Colorful Tree (2017 多校第一场 1003) 【树形dp】 - wy_2016 - 博客园
题意:一条链的权值定义为链上颜色的数量,求给定的树中所有不同链的权值之和。
一种颜色对答案的贡献为包含这种颜色的链的条数。因此可以对每种颜色求不包含这种颜色的链的条数。
sum[c] 表示已经dfs过的所有点中,为颜色 c 的子树的点的数量。
当dfs到