HDU 1520 Anniversary party


树形DP第一题。首先定义dp[i][0]表示不选i号节点i子树能获取的最大值,dp[i][1]表示选i号节点i子树能获取的最大值。那么很容易的可以写出状态转移方程:

dp[i][0]+=max(dp[v][0],dp[v][1]);
dp[i][1]+=dp[v][0];


#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=6010;
vector<int> g[maxn];
int val[maxn],in[maxn],dp[maxn][2];

int dfs(int rt)
{
    dp[rt][0]=0;
    dp[rt][1]=val[rt];
    int len=g[rt].size();
    for(int i=0;i<len;i++)
    {
        int v=g[rt][i];
        dfs(v);
        dp[rt][0]+=max(dp[v][0],dp[v][1]);
        dp[rt][1]+=dp[v][0];
    }
    return max(dp[rt][0],dp[rt][1]);
}

int main()
{
    int n,l,k;
    while(~scanf("%d",&n))
    {
        memset(in,0,sizeof in);
        for(int i=1;i<=n;i++) scanf("%d",&val[i]),g[i].clear();
        while(~scanf("%d%d",&l,&k))
        {
            if(l==0&&k==0) break;
            g[k].push_back(l);
            in[l]++;
        }
        for(int i=1;i<=n;i++)
        {
            if(!in[i])
            {
                printf("%d\n",dfs(i));
                break;
            }
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值