树形DP入门

树结构和DP的结合
一般分为以下几个步骤
1.建立树(通常为有根树)
2.遍历树(可以结合DFS也可以用记忆化搜索)
3.DP
通常使用的DP状态dp[i][0],dp[i][1]代表了第i个节点选用和不选用的状态。
题目地址
以HDU1520为例,说明DP状态
设dp[i][j]表示这个节点选用或不选用的最优解
我们发现每个节点都有选和不选的状态,如果第i个节点选了那么就有
dp[i][1]+=dp[i+1][0];
否则
dp[i][0]+=max(dp[i+1][1],dp[i+1][0]);
其中我们视i+1为i的son节点

#include <bits/stdc++.h>

using namespace std;
int value[6010],dp[6010][2],father[6010];
vector <int> son[6010];
void DP(int t)
{
    dp[t][0]=0;
    dp[t][1]=value[t];
    for(int i=0;i<son[t].size();i++)
    {
        int son1=son[t][i];
        DP(son1);//此处注意,顺序不能在下面两个dp前面,毕竟你只知道 dp[t][0]=0; dp[t][1]=value[t];
        //你需要知道他的子节点的值,实际上会一直dfs到这条树的最低端。
        dp[t][0]+=max(dp[son1][1],dp[son1][0]);
        dp[t][1]+=dp[son1][0];
    }
}
int main()
{
    freopen("in.txt","r",stdin);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>value[i];
        father[i]=-1;//记住初始化节点关系
        son[i].clear()//这句其实没有也不影响,但是思路上来讲是需要的。
    }
    int a,b;
    while(cin>>a>>b&&a!=0&&b!=0)
    {
        son[b].push_back(a);//因为一个父节点有多个子节点,一个子节点有1个父节点,所以用这种方式
        father[a]=b;
    }
    int t=1;
    while(father[t]!=-1)
        t=father[t];
    DP(t);
    cout<<max(dp[t][0],dp[t][1])<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值