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;
}