hdu1520 数形dp入门

最新推荐文章于 2021-05-20 19:36:11 发布

Mizersy

最新推荐文章于 2021-05-20 19:36:11 发布

阅读量256

点赞数

分类专栏：动态规划

本文链接：https://blog.csdn.net/Mizersy/article/details/79138513

版权

动态规划专栏收录该内容

4 篇文章 0 订阅

订阅专栏

这是一道关于树形动态规划的题目，目标是在一棵有权值的有根树中，从每个节点中选择一个，使得所选节点的权值之和最大。解决方案中，定义dp[i][0]表示节点i不被选，dp[i][1]表示节点i被选的最优解，并通过转移方程更新状态，确保每个节点的子节点中仅有一个被选中。

摘要由CSDN通过智能技术生成

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

思路：
树形dp的常规入门题：设dp[i][0]表示:当前这个点不选，dp[i][1]表示当前这个点选择的最优解。

转移方程：dp[cur][0]+=max(dp[son][1],dp[son][0]);//当前这个点不选，那他的孩子可选可不选，取
最大的。

dp[cur][1]+=dp[son][0]//当前这点选择，那他的孩子就不能选择。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
const int N = 6010;
int f[N],happy[N],n;
int dp[N][2];
vector <int> E[N];
void DFS(int t){
    dp[t][1] = happy[t];
    for (int i = 0;i < E[t].size();i++){
        int son = E[t][i];
        DFS(son);
        dp[t][1] += dp[son][0];
        dp[t][0] += max(dp[son][0],dp[son][1]);
    }
}
int main(){
    while(~scanf("%d",&n)){
        for (int i = 1;i <= n;i++){
            scanf("%d",&happy[i]);
            f[i] = -1;
            E[i].clear();
        }
        memset(dp,0,sizeof(dp));
        int L,K;
        while(scanf("%d%d",&L,&K) && (L+K)){
            f[L] = K;
            E[K].push_back(L);
        }
        int t = 1;
        while(f[t]!= -1) t = f[t];
        DFS(t);
        printf("%d\n",max(dp[t][1],dp[t][0]));
    }
    return 0;
}