P1352 没有上司的舞会(树形DP）

1352

题目描述 某大学有 n 个职员，编号为 1-n。

他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。

现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri
，但是呢，如果某个职员的直接上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。

所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

我们可以定义 dp[i][0/1] 代表以 i 为根的子树的最优解（第二维的值为 0 代表 i 不参加舞会的情况，1 代表 i 参加舞会的情况）。

显然，我们可以推出下面两个状态转移方程（其中下面的 x 都是 i 的儿子）：

dp[i][0] += max{dp[x][1],dp[x][0])（上司不参加舞会时，下属可以参加，也可以不参加）
dp[i][1] += dp[x][0]（上司参加舞会时，下属都不会参加）
这里我们定义一个adj的vector数组，来存储职员之间的关系，其中第一维存储的是上司，第二维大小是可变的，存储其对应的下属。happy[i]表示职员 i 快乐值，vis[i]表示职员 i 是否有上司。

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 6005;
int happy[maxn], vis[maxn];
vector<int> adj[maxn];
int dp[maxn][2];
void solve(int x)
{
    dp[x][0] = 0;
    dp[x][1] = happy[x];
    for (int i = 0; i < adj[x].size(); i++)
    {
        int y = adj[x][i];
        solve(y);
        dp[x][0] += max(dp[y][0], dp[y][1]);
        dp[x][1] += dp[y][0];
    }
}
int main()
{
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> happy[i];
    }
    int x, y;
    for (int i = 1; i < n; i++)
    {
        cin >> x >> y;
        if (!x && !y)
            break;
        adj[y].push_back(x);
        vis[x] = 1;
    }
    int root = -1;
    for (int i = 1; i <= n; i++)
    {
        if (!vis[i])
        {
            root = i;
            break;
        }
    }
    solve(root);
    cout << max(dp[root][0], dp[root][1]) << endl;
    return 0;
}