A - Anniversary party HDU - 树形DP

最新推荐文章于 2020-05-02 17:00:02 发布

ID_BePosit

最新推荐文章于 2020-05-02 17:00:02 发布

阅读量213

点赞数

CC 4.0 BY-SA版权

分类专栏： DP

本文链接：https://blog.youkuaiyun.com/BePosit/article/details/83279405

DP 专栏收录该内容

105 篇文章

订阅专栏

本文介绍了一种使用树形动态规划(DP)解决公司晚会最优邀请方案的问题，通过构建员工之间的上下级关系图，利用DFS遍历并递归回溯，实现了在不邀请直接上司的前提下最大化晚会氛围值的算法。

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

A - Anniversary party
HDU - 1520
树形DP-模板
题意：有个公司要举行一场晚会。为了让到会的每个人不受他的直接上司约束而能玩得开心.
公司领导决定：如果邀请了某个人，那么一定不会再邀请他的直接的上司，但该人的上司的上司.
上司的上司的上司……都可以邀请。已知每个人最多有唯一的一个上司。
已知公司的每个人参加晚会都能为晚会增添一些气氛，求一个邀请方案，使气氛值的和最大
树形DP DFS递归到叶子节点开始回溯，父亲去只能由儿子不去来更新，父亲不去则可以在儿子去
或不去中选择一个最大值。

#include<bits/stdc++.h>
using namespace std;
#define maxn 6666
int happy[maxn],pre[maxn];
int dp[maxn][maxn],a,b,n;
vector<int>edge[maxn];
void dfs(int cur)
{
    dp[cur][1]=happy[cur];
    int len =edge[cur].size();
    for(int i=0; i<len; i++)
    {
        int son=edge[cur][i];
        dfs(son);
        dp[cur][0]+=max(dp[son][1],dp[son][0]);
        dp[cur][1]+=dp[son][0];
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&happy[i]);
            edge[i].clear();
            dp[i][0]=dp[i][1]=0;
            pre[i]=-1;
        }
        while(1)
        {
            scanf("%d%d",&a,&b);
            if(a==0&&b==0)break;
            edge[b].push_back(a);
            pre[a]=b;
        }
        int root=1;
        while(pre[root]!=-1)root=pre[root];
        dfs(root);
        printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0;
}