题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=4616
题目大意:
给一棵树,每个节点有一个礼物值及是否有trick,每来到一个节点必须拿礼物,如果该节点有trick则浪费一个抗trick机会,如果一开始有C次抗trick的机会,问最多可以拿多少值的礼物,机会用完后就结束了。访问了的节点不能再次访问。
解题思路:
树形dp啊。
树形dp很不熟悉阿阿阿阿阿阿阿阿。。。。
dp[i][j][0]表示从节点i开始,有j次机会,在子树中最大的能拿到礼物的值。
dp[i][j][1]表示从节点i开始,有j次机会,在子树中次大的能拿到礼物的值。
by[i][j]表示i节点,有j次机会时,得到最大值的那个直接儿子标号。
先以任意一点为根,dfs一遍。求出从每个节点的开始的在子树中的最大值和次大值。
再以同一点为根,dfs一遍,参数维护一个以父亲为开始节点且不经过该节点的最大值。求出该节点出发的在整棵树中的最大值。
PS:
注意此题麻烦的地方是,当机会恰好用完时,游戏结束,不能累加后面的0机会值。
所以要分该节点是否有trick来判断,如果有trick的话,只有j>=2时有从后面更新。
PS2:要多做树形dp啊。
代码:
#include
#include
#include
#include
#include
#include
#include
#include
#include