hdu1520 树形dp

树形dp入门，在树上进行dp。

状态转移方程:

dp[i][0] = max(dp[j][0], dp[j][1]);//i为j的上司 并且i不来

dp[i][1] = dp[j][0];//i来了

用vector实现 rt表示树。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int MAXN = ;
vector<int> rt[MAXN];
int dp[MAXN][],n,vis[MAXN], pa[MAXN];
int max(int x,int y){
    return x>y?x:y;
}
void tree_dp(int root)
{
    vis[root] = ;
    int len = rt[root].size();
    for(int i=; i<len; i++){
        if(!vis[rt[root][i]]){
            tree_dp(rt[root][i]);
            dp[root][] += dp[rt[root][i]][];
            dp[root][] += max(dp[rt[root][i]][], dp[rt[root][i]][]);
        }
    }
}
int main()
{
    int i;
    while(~scanf("%d",&n))
    {
        memset(dp,,sizeof(dp));
        memset(vis,,sizeof(vis));
        memset(pa,-,sizeof(pa));
        for(i=; i<=n; ++i){
            int v;
            rt[i].clear();
            scanf("%d",&v);
            dp[i][] = v;
        }
        int x,y;
        while(scanf("%d%d",&x,&y)){
            if(!x && !y)break;
            pa[x] = y;
            rt[y].push_back(x);
        }
        int root = ;
        while(pa[root]!=-)
            root = pa[root];
        tree_dp(root);
        printf("%d\n",max(dp[root][], dp[root][]));
    }
}