【树型dp】没有上司的舞会

树型dp是动态规划里一种常见题型，顾名思义，树型dp在题目中很明显就能看出来是一棵树，而且需要用到dp的思想来做。下面用一道例题来举例并解释何为树型dp。

例题：没有上司的误会

原题 acwing 没有上司的舞会

Ural 大学有 N 名职员，编号为 1∼N。
他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。
每个职员有一个快乐指数，用整数 $H_i$ 给出，其中 1≤i≤N。
现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。
在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。
输入格式
第一行一个整数 N。
接下来 N 行，第 i 行表示 i 号职员的快乐指数 $H_i$。
接下来 N−1 行，每行输入一对整数 L,K，表示 K 是 L 的直接上司。（注意一下，后一个数是前一个数的父节点，不要搞反）。
输出格式
输出最大的快乐指数。
数据范围
1≤N≤6000,
−128≤$H_i$≤127
输入样例：
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出样例：
5

大致解释一下题意：就是给一棵树，然后要在里面选一些点，其中这些点不能有直接的父子关系，然后要找到点数总和最大的选法。
其实这道题目也可以算是深度遍历，但是要考虑到如何表示某个人参不参加的状态以及如何用其他状态推出当前人的状态，所以也划归为dp问题。

那么先从状态表示入手。用$f(i,1)$表示以$i$为根节点，$i$参加了舞会的所有方案，$f(i,0)$表示以$i$为根节点，$i$没有参加舞会的所有方案，集合的属性自然就是以$i$为根节点，这棵子树快乐的最大值。

下一步思考状态计算。在第一步已经划分好了集合，那么就要考虑怎么计算。这里使用递归明显是更好理解的。对于$f(i,j)$，遍历$i$的所有出边，递归求出以$i$的子节点为根节点的子树的最大快乐值。用$j$表示$i$的子节点，这时又分成两种情况：

1. 如果$j=1$，那么代表$i$参加了舞会，所以$i$的子节点就不能参加舞会了，那么$f(i,j)+=f(j,0)$；
2. 如果$j=0$，那么代表$i$没参加舞会，所以$i$的子节点既可以参加舞会，也可以不参加，那么$f(i,j)+=max(f(j,0),f(j,1))$。

最后要处理一些细节问题：

1. 本题为稀疏图，用邻接表存。不懂邻接表的同学可以看一下C++：用邻接矩阵和邻接表存图；
2. 没给出总的根节点，要自己判断一下。也就是找到没有父节点的节点即可；
3. 最后输出时，要输入$f(root,1or0)$，代表整棵树的快乐最大值。

题解：

#include <cstring>
#include <iostream>

using namespace std;

const int N = 6010;

int h[N], e[N], ne[N], idx;
int happy[N];
int f[N][2];
bool has_fa[N];
int 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; i = ne[i])
    {
        int j = e[i];
        //先递归算出所有子树的快乐最大值
        dfs(j);
        
        f[u][1] += f[j][0];
        f[u][0] += max(f[j][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);
        //标记：a有父节点
        has_fa[a] = true;
    }
    
    //找到总的根节点
    int root = 1;
    while (has_fa[root]) root ++ ;
    
    dfs(root);
    
    cout << max(f[root][0], f[root][1]) << endl;
    
    return 0;
}