【树形DP】hdu 1520

dp[i][0]表示不取该点，dp[i][1]取该点

转移方程：dp[i][1] = w[i] + ∑dp[j][0];

                    dp[i][0] = ∑max(dp[j][0] , dp[j][1]);(j为i的儿子)

 

#include <map>
#include <set>
#include <list>
#include <queue>
#include <deque>
#include <stack>
#include <string>
#include <time.h>
#include <cstdio>
#include <math.h>
#include <iomanip>
#include <cstdlib>
#include <limits.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;

#define LL long long
#define MIN INT_MIN
#define MAX INT_MAX
#define PI acos(-1.0)
#define N 6010
#define FRE freopen("a.txt","r",stdin)
int max(int a,int b){return a>b?a:b;}
int dp[N][2];
int w[N];
bool vis[N];
vector<int> v[N];
int n;
void init(){
    int i;
    for(i=0;i<=n;i++){
        v[i].clear();
        w[i]=0;
        vis[i]= true;
    }
    memset(dp,0,sizeof(dp));
}
void dfs(int r){
    int sz = v[r].size();
    if(sz==0){
        dp[r][0] = 0;
        dp[r][1] = w[r];
        return ;
    }
    int i;
    for(i=0; i<sz; i++){
        int x = v[r][i];
        dfs(x);
        dp[r][0]+= max(dp[x][0], dp[x][1]);
        dp[r][1]+= dp[x][0];
    }dp[r][1]+= w[r];
}
int main(){
    while(scanf("%d",&n)!=EOF){
        int i,j;
        init();
        for(i=1;i<=n;i++)
        scanf("%d",&w[i]);
        int a,b;
        while(scanf("%d%d",&a,&b) &&(a+b)){
            v[b].push_back(a);
            vis[a] = false;
        }
        int root;
        for(i=1;i<=n;i++)
            if(vis[i]){
                root = i;
                break;
            }
        dfs(root);
        int ans = MIN;
        for(i=1;i<=n;i++){
            ans = max(ans,dp[i][0]);
            ans = max(ans,dp[i][1]);
        }
        printf("%d\n",ans);
    }
    return 0;
}