树形DP一般用于一些树形结构的题目,而且是算法竞赛中比较喜欢考察的一个知识点。
树形DP入门
例题1:AcWing 285.没有上司的舞会
这题非常简单,可称之为树形DP的入门题。显然,一个节点有两种状态:选(参加)或不选(不参加)。那么不妨设fu,1表示节点u被选后u节点及其子树可获得的最大权值,fu,0表示节点u不被选的情况下,u节点及其子树可获得的最大权值。那么很容易可以推出:fu,1(+=fson,0)+wu(其中son表示u的所有子节点)。fu,0+=max(fson,1,fson,0)。那么这个DP顺序显然要按照dfs的顺序来按深度遍历,而且要用递归的方式先求出子节点的状态。所以代码就很容易写了:
#include<iostream>
using namespace std;
const int N=6005;
int h[N],f[N][2];
int head[N],to[N],ne[N],idx;
bool hfa[N];
void add(int x,int y){
ne[++idx]=head[x];
to[idx]=y;
head[x]=idx;
}
void dfs(int u){
f[u][1]=h[u];
for(int i=head[u];i;i=ne[i]){
int j=to[i];
dfs(j);
f[u][0]+=max(f[j][0],f[j][1