树形DP
树形DP是一种优美的动态规划
一.原理分析
为什么可以在树上做DP呢?
首先,树结构本身就具有最优子结构,当前节点的答案可以向子节点传递/子节点可以改变父节点答案.
其次,树形结构能处理一些特殊的关系,如LuoguP1352 没有上司的舞会,每个人与其上司都有一种依赖的关系,而众多依赖关系就形成一颗树.
二.状态转移方程与数组状态设计
DP最难的地方,想必就是求状态转移方程了叭,虽然每道题的状态转移方程都不一样,但是多多少少我们能从中找到一些规律.
1.状态设计
通常,状态设计为二维dp[x][ ],第一维的x表示当前节点为x,也可以理解为以x节点为根的子树所求的最优答案,而后面一维则需视情况而定.
2.实现形式
因为是树形结构嘛,所以理所当然的用dfs递归来实现啦,实现时一般都是用子树更新父亲(即从下向上更新),那么现在就有个问题惹:首先应该考虑的是一个一个子树的更新父亲还是把所有子树都算完了再更新父亲?
这就要因题而异了,一般来说有两种情况:1.需要把所有子树的信息都掌握之后再更新父亲的就需要把所有子树都算完了在更新父亲。2.而像树上背包这样的问题就需要一个一个的更新,每次都用一个子树更新已经更新完的子树+父亲,最后就可以将这一部分的子树更新完了,再继续往上更新,最后根节点就是答案。
其实上面的两种情况可以总结成一种情况就是一个个子树更新父亲,一般来说第一种情况应用更多,也能解决第二情况的问题,只不过如果符合第二种情况的时候用第二种可以速度更快一点,毕竟你省了一遍循环嘛。
(这两段转自dcx2001)
如果不懂的话没有关系,后面我会用具体题目来分析.
三.模板题
1.LuoguP1352 没有上司的舞会
题目描述
某大学有 n 个职员,编号为 1 … \ldots … n。
他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。
现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数 rir_iri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。
所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
输入格式
输入的第一行是一个整数 nnn。
第 2 到第 ( n + 1 ) (n+1) (n+1) 行,每行一个整数,第 ( i + 1 ) (i+1) (i+1) 行的整数表示 i i i 号职员的快乐指数 r i r_i ri。
第 ( n + 2 ) (n + 2) (n+2) 到第 2 n 2n 2n 行,每行输入一对整数 l , k l,k l,k