树形DPhdu1502

题目描述:每个节点有权值，不能同时选择相邻的两个节点，问最后选的最大价值。
贪心的想法是要不选子节点，要不选父亲节点，可是有可能最优的是两个节点都不选。
那么就是类似01背包选择路线，来个dp[i][2],0表示不选，1表示选，那么从叶子节点往根节点不断更新就好了，状态转移方程
dp[i][1]=sum(dp[j][0])
dp[i][0]=sum(max(dp[j][1].dp[j][0])

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<vector>
using namespace std;
vector<int>Edge[6010];
int visit[6010],dp[6010][2],val[6010];

void dfs(int u)
{
    if(visit[u]) return;
    visit[u]=1;
    for(int i=0;i<Edge[u].size();i++)
    {
        int v=Edge[u][i];
        if(!visit[v])
        {
            dfs(v);
            dp[u][0]+=max(dp[v][1],dp[v][0]);
            dp[u][1]+=dp[v][0];
        }
    }
    dp[u][1]+=val[u];
}

void init()
{
    memset(dp,0,sizeof(dp));
}

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        scanf("%d",&val[i]);
        for(int i=1;i<=n;i++)
        {
            Edge[i].clear();
            visit[i]=0;
        }
        int a,b;
        scanf("%d %d",&a,&b);
        while(a+b!=0)
        {
            Edge[a].push_back(b);
            Edge[b].push_back(a);
            scanf("%d %d",&a,&b);
        }
        init();
        dfs(1);
        printf("%d\n",max(dp[1][0],dp[1][1]));
    }
}