poj 2342

第一把树形dp ， 不懂树形dp，百度学习，这是树形dp入门，对着别人的代码打的，然后懂了。
题意：有一个公司的人要去参加party，每个人有一个愉悦值。每个人和他的直接上司不能同时出现。要求邀请一定量的人使得愉悦值最大。

树形dp，对于第i个人，有两种选择，选或不选。dp[i][0] , dp[i][1];

推出dp方程dp[i][0] += max(dp[st][1],dp[st][0]) st为i的所用子树。
dp[i][1] += d[st][0]; st为子树。
对根节点做一遍的dfs就可以求出最大值了。 代码更加容易懂，上代码。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
#define maxn 10010
int dp[maxn][2];
int father[maxn],visit[maxn];
int n;
void dfs(int root) {
    visit[root] = 1;
    for(int i = 1; i <= n; i++) {
        if(visit[i] == 0 && father[i] == root) {
            dfs(i);
            dp[root][0] += max(dp[i][0],dp[i][1]);
            dp[root][1] += dp[i][0];
        }
    }
}
int main() {
    while(~scanf("%d",&n)) {
        memset(dp,0,sizeof(dp));
        memset(visit,0,sizeof(visit));
        memset(father,0,sizeof(father));
        for(int i = 1; i <= n; i++)
          scanf("%d",&dp[i][1]);
        int x,f,root;
        while(scanf("%d %d",&x,&f),x||f) {
            father[x] = f;
        }
        root = x;
        while(father[root]) root = father[root];

        dfs(root);
        printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0;
}