[树形dp] 没有上司的舞会(模板题+树形dp)

0. 前言

树形 dp 思维门槛高，但是跨过门槛之后，思维难度却不高。这点非常类似于

关键是要体会采用二进制数来表示状态的思想，要转变传统思维，学习接收并吸收这种思想。

1. 树形dp 模板题

285. 没有上司的舞会

简单模拟样例，便于理解：

样例的高兴度都是 1，那就是挑选没有相邻边的尽量多的点

---

树形 dp 思路：

重点： 状压 dp

思路：

• 状态定义：
  • f[u][0]：所有从以 u 为根的子树中选择，并且不选 u 这个点的方案的最大高兴度
  • f[u][1]：所有从以 u 为根的子树中选择，并且选 u 这个点的方案的最大高兴度
• 状态转移：
  • 分类依据：求解树形 dp 问题，大多都是递归来求解。即当求解根节点的时候，其所有儿子的状态都已经确定了。假设当前求解根节点为 $u$，其所有儿子节点为 $s_i$。那么针对根节点的选法进行分类：
    • 不选根节点，即 $f[u][0]=\Sigma (f[s_i][0],f[s_i][1])$。其含义为：不选根节点时，其任一儿子节点选与不选均可，在两者中取最大值将其累加起来即可
    • 选根节点，即 $f[u][1]=\Sigma (f[s_i][0])$。其含义为：选根节点时，其任一儿子节点均不能选，直接累加起来即可
• 状态转移方程：$f[u][0]=\Sigma (f[s_i][0],f[s_i][1])$、$f[u][1]=\Sigma (f[s_i][0])$
• 状态初始化：首先需要建图，邻接表的头节点初始化为 -1。一开始当前节点去的话，则初始高兴度初始化为它自己 f[u][1]=happy[u]，不去的话 f[u][0]=0
• 返回答案： f[root][0], f[root][1]
• 时间复杂度： 一共有 $2n$ 个状态，每个状态在计算过程中需要枚举它所有的儿子，则所有节点儿子的数量总和就等于树中边的数量，边的数量是 $n-1$。则在状态计算时，故计算所有状态就是 $O(n)$

这个 dfs(root) 根节点是需要的，并不能像之前一样，随意找一个点进行 dfs。因为建图的时候采用的是单向边。

本题的物理结构相当于是一颗有根的树，其不适用线性 dp 的递推方式。而树形 dp 的这种递归求解的方式是恰好适用本题。

动态规划的本质是沿着一个拓扑序来确定状态的值的，对于一个树形结构而言，子节点的值远比根节点的值好确定，那么就是以叶子节点的值层层向上确定其父节点、根节点的值。这就是递归，且遵循拓扑序。所以采用 dfs 进行递归，且在回溯的时候进行状态计算，因为在回溯的时候，其子节点的状态全部计算完毕了。

本题非常非常非常经典，是树形 dp 最经典题目。

代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 6005;

int n;
int happy[N];
int h[N], e[N], ne[N], idx;
int f[N][2];
bool has_father[N];

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++; 
}

void dfs(int u) {
    f[u][1] = happy[u];
    
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = e[i];
        dfs(j);
        
        f[u][0] += max(f[j][0], f[j][1]);
        f[u][1] += f[j][0];
    }
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; ++i) cin >> happy[i];

    memset(h, -1, sizeof h);
    for (int i = 0; i < n - 1; ++i) {
        int a, b;
        cin >> a >> b;
        add(b, a);
        has_father[a] = true;
    }
    
    int root = 1;
    while (has_father[root]) root ++;
    
    dfs(root);
    
    cout << max(f[root][0], f[root][1]) << endl;
    
    return 0;
}