DP问题——没有上司的舞会（树形DP典型例题）

状态表示：f[i][1/0]表示以i为根节点的树包含/不包含根节点时的最大快乐值

状态划分：以是否包含根节点为标准划分

包含根节点时 则一定不含根节点的子节点 。即f[i][1] = v[i] + f[子节点][0]

（v[i]表示节点i的快乐值）

不包含根节点时 则根节点的子节点不一定包含。即f[i][0] =

算出所有情况后，取max(f[root][0],f[root][1])

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 6010;
int v[N],e[N],ne[N],idx,n;
int h[N];
int f[N][2];
bool flag[N];
 

void add(int a,int b){
    e[idx] = b;ne[idx] = h[a];h[a] = idx++;
}
void dfs(int x){
    f[x][1] = v[x];
    
    for(int i = h[x];i != -1;i = ne[i]){
        int a = e[i];
        dfs(a);
    f[x][0] += max(f[a][0],f[a][1]);
    f[x][1] += f[a][0]; 
    }
}
int main(){
    cin >> n;
    for(int i = 1;i <= n;i++) cin >> v[i];
    memset(h,-1,sizeof h);
    for(int i = 1;i < n;i++) {
        int x,y;
        cin >> x >> y;
        add(y,x);
        flag[x] = true;
    }
    int i = 1;
    while(flag[i]) i++;
    dfs(i);
    
    cout << max(f[i][0],f[i][1]);
    return 0; 
}

总结：树形DP首先要构造树——把树视为特殊的图来构造。利用add函数。

           其次树形DP枚举所有情况不是两个for循环了，本题中为dfs，要会灵活使用不同的遍历方法。