题意:某大学有n个职员,编号为1~n。他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数Ri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
题解:树形dp
设
d
p
[
x
]
[
0
]
dp[x][0]
dp[x][0]表示以x为根的子树,且x不参加舞会的最大快乐值;
d
p
[
x
]
[
1
]
dp[x][1]
dp[x][1]表示以x为根的子树,且x参加了舞会的最大快乐值。则转移方程为:
d
p
[
x
]
[
0
]
+
=
m
a
x
(
d
p
[
y
]
[
0
]
,
d
p
[
y
]
[
1
]
)
,
d
p
[
x
]
[
1
]
+
=
d
p
[
y
]
[
0
]
+
h
[
x
]
dp[x][0] += max(dp[y][0], dp[y][1]),dp[x][1] +=dp[y][0] + h[x]
dp[x][0]+=max(dp[y][0],dp[y][1]),dp[x][1]+=dp[y][0]+h[x],其中y是x的下属。
以root为根节点,递归子结点即可,最终答案为 m a x ( d p [ r o o t ] [ 0 ] , d p [ r o o t ] [ 1 ] ) max(dp[root][0], dp[root][1]) max(dp[root][0],dp[root][1])。
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
#define maxn 6666
int h[maxn], f[maxn], dp[maxn][2], n, x, y, root;
vector<int> son[maxn];
void dfs(int x) {
dp[x][0] = 0;
dp[x][1] = h[x];
int len = son[x].size();
for (int i = 0; i < len; i++) {
int y = son[x][i];
dfs(y);
dp[x][0] += max(dp[y][0], dp[y][1]);
dp[x][1] += dp[y][0];
}
}
int main() {
while (~scanf("%d", &n)) {
memset(dp, 0, sizeof(dp));
memset(son, 0, sizeof(son));
for (int i = 1; i <= n; i++) {
scanf("%d", &h[i]);
}
while (scanf("%d%d", &x, &y) && x && y) {
son[y].push_back(x);
f[x] = y;
}
root = 1;
while (f[root]) {
root = f[root];
}
dfs(root);
printf("%d\n", max(dp[root][0], dp[root][1]));
}
return 0;
}