HDU 1520 Anniversary party(树形DP)

最新推荐文章于 2019-06-10 19:50:54 发布

原创最新推荐文章于 2019-06-10 19:50:54 发布 · 665 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#HDU #dp

DP 专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一个关于树形动态规划的入门级算法，详细解释了如何通过递归实现从叶子节点到根节点的DP过程，最终输出根节点的最大rank。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1520

这个应该算是树形DP入门级的题目了，比较简单，每个节点设置两个值，一个是my是选择自己时的rank值最大，一个是other

表示不选择自己时的rank最大，然后从叶子节点开始DP（从根递归实现），最后输出根节点my 和 other中较大的一个

#include <iostream>
#include <string.h>
#include <algorithm>
#include <stdio.h>
#include <vector>
using namespace std;
#define maxn 10000
#define MAX(a,b) (a>b?a:b)
struct node{
    int rank;
    int my,other;
    vector<int> child;
}po[maxn];
int n;
bool is_root[maxn];
int tree_dp(int root){
    if(po[root].child.empty()){
        po[root].my=po[root].rank;
        po[root].other=0;
        return 0;
    }
    for(int i=0;i<po[root].child.size();i++){
        tree_dp(po[root].child[i]);
    }
    int ans1=0,ans2=0;
    for(int i=0;i<po[root].child.size();i++){
        ans1+=MAX(po[po[root].child[i]].my,po[po[root].child[i]].other);
        ans2+=po[po[root].child[i]].other;
    }
    po[root].my=ans2+po[root].rank;//表示选自己就一定不能选其子节点的任意一个，所以所有子节点都要选other
    po[root].other=ans1;//表示不选自己
    return 0;
}
int main(){
    int i,j,k,a,b,root;
    while(scanf("%d",&n)!=EOF){
        for(i=1;i<=n;i++){
            scanf("%d",&po[i].rank);
            po[i].my=po[i].other=0;
            po[i].child.clear();
        }
        memset(is_root,true,sizeof(is_root));
        while(scanf("%d%d",&a,&b),a+b){
            po[b].child.push_back(a);
            is_root[a]=false;
        }
        for(i=1;i<=n;i++)
        if(is_root[i]){
            root=i; break;
        }
        tree_dp(root);
        printf("%d\n",MAX(po[root].my,po[root].other));
    }
    return 0;
}