poj 2342 树形DP

同poj 3342

http://www.cnblogs.com/wuyiqi/archive/2012/03/08/2385680.html

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
#include<map>
using namespace std;
#define max(a,b) a>b?a:b
const int maxn  =  6010;
vector<int> edge[maxn];
int dp[6010][2];
int w[6010];
int n;
void dfs(int u,int p)
{
    int i,j;
    dp[u][1]=w[u];
    dp[u][0]=0;
    for(i=0;i<edge[u].size();i++)
    {
        int v=edge[u][i];
        if(v==p) continue;
        dfs(v,u);
        dp[u][1]+=dp[v][0];
        dp[u][0]+=max(dp[v][0],dp[v][1]);
    }
}
int in[maxn];
int main()
{
    int i,j,k,s,e;
    for(i=0;i<=6000;i++) edge[i].clear();
    scanf("%d",&n);
    for(i=1;i<=n;i++) scanf("%d",&w[i]);
    memset(in,0,sizeof(in));
    while(scanf("%d%d",&e,&s),(e+s)!=0)
    {
        in[e]++;
        edge[s].push_back(e);
    }int flag=-1;
    for(i=1;i<=n;i++)
        if(in[i]==0)
        {
            flag=i;
            break;
        }
    dfs(flag,flag);
    printf("%d\n",max(dp[flag][0],dp[flag][1]));
    return 0;
}