hdu 1520

树形dp的基础题；

状态转移很简单：老爸没选，儿子可选可不选，最大就行；

老爸选了，儿子肯定不能选；

dp[root][0]+=max(dp[son][1],dp[son][0]);
dp[root][1]+=dp[son][0];

代码：

#include<cstdio>
#include<cstring>
#include<vector>
#define maxn 6003
using namespace std;

int w[maxn],dp[maxn][2];
vector<int>ve[maxn];
bool vis[maxn];

void dfs(int a)
{
    vis[a]=1;
    dp[a][1]=w[a];
    dp[a][0]=0;
    int l=ve[a].size();
    for(int i=0; i<l; i++)
    {
        int v=ve[a][i];
        if(vis[v]) continue;
        dfs(v);
        dp[a][0]+=max(dp[v][1],dp[v][0]);
        dp[a][1]+=dp[v][0];
    }
}

int main()
{
    int n,a,b;
    while(scanf("%d",&n)!=EOF)
    {
        memset(vis,0,sizeof vis);
        memset(dp,0,sizeof dp);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&w[i]);
            ve[i].clear();
        }
        while(scanf("%d%d",&a,&b)&&(a+b))
        {
            ve[a].push_back(b);
            ve[b].push_back(a);
        }
        dfs(1);
        printf("%d
",max(dp[1][0],dp[1][1]));
    }
    return 0;
}