HDU 1520 Anniversary party | 树形dp

水了好几天，还得准备开学那篇实习心得。。。顶水的 = =

题意：

邀请一些人作客，这些人两两可能是直接上下属关系。为了不破环气氛，不能同时邀请有直接上下属关系的人。每个人都有一个兴奋值，让你求出最大的兴奋值。

思路：

经典的树形dp。

状态定义：

dp[i][0]：不要第 i 个人所获得的最大的兴奋值。

dp[i][1]：要第 i 个人所获得最大兴奋值。

状态转移方程：( v 为 i 的直接儿子）；

dp[i][0] += max（dp[v][1]， dp[v][0]）

dp[i][1] += dp[v][0];

注意：

1.本题还是以文件结束为标志。

2.所谓的0 0结束输入，事实是输入的边肯定是n-1条，否则无法构成连通的一棵树。

AC代码：

#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
const int MAXN = 6005;
int dp[MAXN][2], val[MAXN];
vector <int> gra[MAXN];
int ind[MAXN], n;
void dfs(int u)
{
        dp[u][1] = val[u], dp[u][0] = 0;
        for(int i = 0;i < gra[u].size(); i++)
        {
                int v = gra[u][i];
                dfs(v);
                dp[u][1] += dp[v][0];
                dp[u][0] += max(dp[v][1], dp[v][0]);
        }
}
int main()
{
        while(scanf("%d", &n) != EOF)
        {
                memset(ind, 0, sizeof(ind));
                for(int i = 1;i <= n; i++)
                        gra[i].clear();
                //
                for(int i = 1;i <= n;i ++)
                        scanf("%d", &val[i]);
                int u, v;
                for(int i = 1;i <= n; i++)
                {
                        scanf("%d%d" ,&v, &u);
                        if(u+v == 0)    continue;
                        gra[u].push_back(v);
                        ind[v]++;
                }
                        
                /*while(scanf("%d%d",&v, &u), u+v)
                {
                        gra[u].push_back(v);
                        ind[v]++;
                }*/
                //
                int res = 0;
                for(int i = 1;i <= n; i++)
                {
                        if(ind[i] == 0)
                        {
                                dfs(i);
                                res = max(dp[i][1], dp[i][0]);
                                break;
                        }
                }
                printf("%d\n", res);
        }
        return 0;
}