POJ 2342 Anniversary party

树形DP入门啦。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

int n,len;
int dp[6001][2];
int value[6001];
bool flag[6001];
vector <int> G[6001];

int max(int a,int b)
{
    if(a > b)   return a;
    return b;
}

void init()
{
    int i;
    for(i = 0;i <= n;i++)
        G[i].clear();
    memset(dp,0,sizeof(dp));
    memset(flag,true,sizeof(flag));
}

void dfs(int v)
{
    for(vector<int>::size_type i = 0;i < G[v].size();i++)
    {
        int u = G[v][i];//u是v的儿子节点
        dfs(u);
        dp[v][0] += max(dp[u][0],dp[u][1]);
        dp[v][1] += dp[u][0];
    }
    dp[v][1] += value[v];
}

int main()
{
    int boss;
    scanf("%d",&n);
    init();
    for(int i = 1;i <= n;i++)
        scanf("%d",&value[i]);
    int p,q;
    while(scanf("%d%d",&p,&q),p + q != 0)
    {
        G[q].push_back(p);
        flag[p] = false;
    }

    for(int i = 1;i <= n;i++)
    {
        if(flag[i])
        {
            boss = i;
            break;
        }

    }
    dfs(boss);

    int ans = max(dp[boss][1],dp[boss][0]);
    printf("%d\n",ans);

    return 0;
}