hdu1520 树形DP

/*
 * 树形DP
 */
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define Max(x,y) (x>y?x:y)
#define max 6000+10

//dp[i][0],dp[i][1],当前节点为i,在父亲去或没去的情况下的最大值
vector<int> adj[max];
int d[max][2],a[max],n;
bool vis[max][2],in[max];

int dp(int i,int j){
    if(vis[i][j]){
        return d[i][j];
    }
    vis[i][j]=true;
    int& ans=d[i][j];

    if(!j){//i的父亲没去那么i可以去
      ans=a[i];
      for(int k=0;k<adj[i].size();k++){
            ans+=dp(adj[i][k],1);
      }
    }

    int sum=0;//任何人都可以选择不去
    for(int k=0;k<adj[i].size();k++){
        sum+=dp(adj[i][k],0);
    }
    ans=Max(ans,sum);
    return ans;
}

int main(){
    while(~scanf("%d",&n)){
        memset(vis,false,sizeof(vis));
        memset(in,false,sizeof(in));
        memset(d,0,sizeof(d));
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            adj[i].clear();
        }
        int a,b;
        while(scanf("%d%d",&a,&b)&&a&&b){
            adj[b].push_back(a); in[a]=true;
        }
        int root;//找到校长，树根
        for(int i=1;i<=n;i++){
            if(!in[i]){
                root=i;break;
            }
        }
        int ans=dp(root,0);
        printf("%d
",ans);
    }
}