题目描述 某大学有 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;
}