没有上司的舞会

本题是经典的树型dp问题。

原题如下：

状态转移方程推导如下：

将状态分为0，1两部分，0为不选择此位，1为选择此位

f[N][2] 前一位代表哪一位，后一位表示是否选择

当选择该位时，其下属则无法参加舞会，

即f[N][1] = sum(f[N的下属][0])

当不选择该位时，其下属则可以选择是否参加舞会，

即f[N][0] = sum(max(f[N的下属][0],f[N的下属][1]))

即可得状态转移方程。

代码如下：

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

using namespace std;
const int N = 6010;
int f[N][2];
int e[N], ne[N], h[N], idx;
int happy[N];
bool have_father[N];
void add(int a, int b)  // 添加一条边a->b
{
    have_father[b] = true;
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

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

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