hdu 1520 Anniversary party（树形dp）

http://acm.hdu.edu.cn/showproblem.php?pid=1520

题意:

在一个有根树上每个节点有一个权值，每相邻的父亲和孩子只能选择一个，问怎么选择总权值之和最大。

思路：

dp[x][0]表示选自己，dp[x][1]表示不选自己，

这样状态转移方程 dp[x][0]+=dp[fa][1],dp[x][1]+=max(dp[fa][1],dp[fa][0]).

从下往上更新就好，

 

代码：

 

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=6050;
const int inf=0x3f3f3f3f;
int n;
vector<int>e[maxn];
int w[maxn];
int in[maxn];
int dp[maxn][2];
void dfs(int x,int pre)
{
    for(int i=0;i<e[x].size();i++)
    {
        int v=e[x][i];
        if(v==pre)continue;
        dfs(v,x);
        dp[x][1]+=max(dp[v][1],dp[v][0]);
        dp[x][0]+=dp[v][1];
    }
}
int main()
{

    while(~scanf("%d",&n))
    {
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            e[i].clear();
            scanf("%d",&w[i]);
        }
        int x,y;
        while(1)
        {
            scanf("%d%d",&x,&y);
            if(x==0&&y==0)break;
            e[y].push_back(x);
            in[x]++;
        }
        for(int i=1;i<=n;i++)
        {
            dp[i][0]=w[i];
        }
        for(int i=1;i<=n;i++)
        {
            if(!in[i])
            {
                dfs(i,0);
                printf("%d\n",max(dp[i][0],dp[i][1]));
                break;
            }
        }
    }
    return 0;
}