ural(Timus) 1039. Anniversary Party

树DP

经典问题，公司聚会，下属和直属上司不能共存，给出每个人的快乐值，再给出每个人的编号和他的上司，选出一些人参加聚会使快乐值最大

/*
从叶子开始选择，每个节点只有选不和不选两种可能，dp[rt][0],dp[rt][1]
分别表示选和不选该节点
dp[rt][1]=sum{ dp[son][0] }+val[rt] , 因为跟选了它的儿子就全部不能选
dp[rt][0]=sum{ max{dp[son][0] , dp[son][1]} }
如果根不选，那么儿子的选择就多样化了，每个儿子又是互不干扰的
所以令每个儿子最优，加起来根就是最优的，而每个孩子无非还是选和不选，一比较就能
得到每个儿子的最优方案
而对于叶子，dp[rt][0]=0;dp[rt][1]=val[rt];
*/

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define N 6010

vector<int>a[N];
int dp[N][2];
int val[N];
int n,root;
bool mark[N],vis[N];

int max(int p ,int q)
{
    return p>q?p:q;
}

void dfs(int rt)
{
    vis[rt]=true;
    int size=a[rt].size();
    dp[rt][0]=0; dp[rt][1]=val[rt];
    for(int i=0; i<size; i++) 
    {
        int son=a[rt][i];
        if(!vis[son])
        {
            dfs(son);
            dp[rt][1] += dp[son][0];
            dp[rt][0] += max(dp[son][0] , dp[son][1]);
        }
    }
}

void solve()
{
    for(int i=1; i<=n; i++)
        if(!mark[i])
        { root=i; break; }
    a[0].push_back(root);
    a[root].push_back(0);
    root=0; val[root]=0;
    memset(vis,false,sizeof(vis));
    dfs(root);
    printf("%d\n",max(dp[root][0] , dp[root][1]));
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        memset(mark,false,sizeof(mark));
        for(int i=1; i<=n; i++) a[i].clear();
        for(int i=1; i<=n; i++) scanf("%d",&val[i]);
        while(1)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            if(!u && !v) break;
            mark[u]=true; 
            a[u].push_back(v);
            a[v].push_back(u);
        }
        solve();
    }
    return 0;
}