个人感觉这是一道比较经典而且基础的树形dp入门题目。
树形dp本质上就是在树的结构上去动态决策。具体到这道题上,可以这样说:对于某个人,有两种决策,让他来或者不让他来,如果让他来,那么他的直接下级肯定不能来;如果不让他来,那么它的直接下级来不来都行。
那么我们就可以得出解法了:首先,从输入中找出根节点(即最高级的上司),从他开始深搜。搜索的内容是以该节点为根节点的子树所能获得的最大开心值。
设dp[i][0]为第i个人不来时以i为根节点的子树的最大开心值,dp[i][1]为第i个人来时的值。j为i的子节点
dp[i][0]=∑max(dp[j][0],dp[j][1]) //第i个人不来时,下级来不来都可以,选开心值大的
dp[i][1]=∑(dp[j][0]) //第i个人来时,他的下级不能来,只能去dp[j][0]
对于具体的代码实现方面,我的习惯是:
(1) 用vector存储某个结点的所有子节点
(2) 在输入时,先设第一个结点为根节点,之后如果某对输入中,子节点为目前的根节点,就把这对输入的根节点设为总的根节点。具体见代码
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int MAXN = 6005;
int dp[MAXN][2];
int happy[MAXN];
vector<int>child[MAXN];
void dfs(int rt)
{
if (child[rt].size() == 0)
{
dp[rt][0] = 0;
dp[rt][1] = happy[rt];
return;
}
int i;
dp[rt][1] = happy[rt]; //要把自己算上
for (i = 0; i < child[rt].size(); ++i)
{
int next = child[rt][i];
dfs(next);
dp[rt][0] += max(dp[next][0],dp[next][1]);
dp[rt][1] += dp[next][0];
}
}
void ini(int n)
{
memset(dp,0,sizeof(dp));
int i;
for (i = 1; i <= n; ++i)
child[i].clear();
}
int main()
{
int n,i,l,k,rt;
while (scanf("%d",&n) != EOF){
ini(n);
for (i = 1; i <= n; ++i)
scanf("%d",&happy[i]);
rt = -1;
while (scanf("%d %d",&l,&k) && l+k){
child[k].push_back(l);
if (rt == -1 || l == rt)
rt = k;
}
dfs(rt);
printf("%d\n",max(dp[rt][0],dp[rt][1]));
}
return 0;
}