多校的第一场的题目。
一个典型的树DP统计题目。这道题目不是难在编程上,而已关键是细节上。
这道题有几个注意的小技巧
1.一开始就要对每个节点的儿子按颜色排个序,这样方便dfs回来的时候统计就按颜色一块块加起来统计就是了,不然的话每次再枚举会高达n^2的复杂度
2.统计的时候主要是根据通过点u连接的两边的方案数*val的总值,稍微注意点细节,很容易算漏算错。算完当前节点,记得算上跟它的父节点的方案,别漏了。
弱菜按着官方题解的提示总算搞完。
附上官方题解:
一次dfs就能搞定的问题,其实没什么复杂的算法,主要难点在于统计时的一些小技巧,个人难度评价是偏难的中档题。
从任意一点开始深搜,每颗子树搜索完毕之后向上返回pair<可以延伸到该点且最后一条边与由父节点到该点的边颜色不同的gorgeous边的条数 , 所有这种边分数的总和>
每次深搜完一个子节点之后,增加的过这一点的gorgeous边的总分数为:
之前深搜的所有子节点向上返回的边数之和 * 当前子节点返回的分数 +
之前深搜的所有子节点向上返回的分数之和 * 当前子节点返回的边数 +
之前深搜的所有子节点向上返回的边数之和 * 当前子节点返回的边数 * 当前点的权
用O(n)的时间把所有的累计起来就好了什么的
才没有呢 = =
如果有当前节点分别到两个子节点的边的颜色相同的话也是不行的,
由于数据中添加了度数很大的点,理论上O(儿子数^2)的两两统计也是要被卡掉的 (希望确实的做到了
正确的做法是对所有的儿子按边的颜色排个序,然后在按这个序深搜和统计