URAL 1039 Anniversary Party

    由于上司的关系构成了一颗树，于是我们可以从树形dp的角度去考虑问题。不妨用f[cur][0]表示不选当前节点时这棵子树value的最大值，f[cur][1]表示选当前节点时这棵子树value的最大值，那么如果不选当前节点，它的子节点就可选也可不选，因此f[cur][0]要加上所有子节点x的Max{f[x][1],f[x][0]}，而如果选当前节点，它的子节点必然不能选，因此f[cur][1]要加上所有子节点x的f[x][0]以及当前节点本身的value。

#include<stdio.h>
#include<string.h>
#define MAXD 6010
int N, first[MAXD], next[MAXD], v[MAXD], e, dgr[MAXD];
int a[MAXD], f[MAXD][2];
int Max(int x, int y)
{
    return x > y ? x : y;    
}
void add(int x, int y)
{
    v[e] = y;
    next[e] = first[x], first[x] = e ++;    
}
void init()
{
    int i, L, K;    
    memset(dgr, 0, sizeof(dgr));
    memset(first, -1, sizeof(first));
    e = 0;
    for(i = 1; i <= N; i ++)
        scanf("%d", &a[i]);
    for(;;)
    {
        scanf("%d%d", &L, &K);
        if(!L && !K)
            break;
        add(K, L), ++ dgr[L];
    }
}
void dp(int cur)
{
    int i, max, sum;
    max = sum = 0;
    for(i = first[cur]; i != -1; i = next[i])
    {
        dp(v[i]);
        max += Max(f[v[i]][0], f[v[i]][1]);
        sum += f[v[i]][0];    
    }
    f[cur][1] = sum + a[cur];
    f[cur][0] = max;
}
void solve()
{
    int i, j, k, ans = 0;
    for(i = 1; i <= N; i ++)
        if(dgr[i] == 0)
        {
            dp(i);
            ans += Max(f[i][0], f[i][1]);    
        }    
    printf("%d\n", ans);
}
int main()
{
    while(scanf("%d", &N) == 1)
    {
        init();
        solve();    
    }
    return 0;    
}