POJ 2342 - Anniversary party （树形dp）

题目链接：

http://poj.org/problem?id=2342

题解：

dp[i][0]表示第i个人不去所获得的最大价值，dp[i][1]表示第i个人去所获得的最大价值。

状态转移方程为：

dp[boss][1] += dp[employee][0]；

dp[boss][0] += max（dp[employee][0]，dp[employee][1]）；

用dfs搜一下就好了。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int dp[6050][2],bos[6040],visit[6050];
int N,boss;
void dfs(int node)
{
for(int i = 1; i <= N; i ++)
{
if(bos[i] == node)
{
dfs(i);
dp[node][1] += dp[i][0];
dp[node][0] += max(dp[i][0], dp[i][1]);
}
}
}

int main()
{
cin >> N;
for(int i = 1; i <= N; i++)
{
int value;
cin >> value;
dp[i][1] = value;
}
int l,k;
while(cin >> l >> k)
{
if(l == 0 && k == 0)break;
bos[l] = k;
boss = k;
}
dfs(boss);
cout << max(dp[boss][0],dp[boss][1]) << endl;;
return 0;
}