HDU - 1520 Anniversary party 树形dp入门+见解

本文探讨了一种宴会邀请问题的动态规划解决方法。给定员工间的领导关系及每位员工的价值,目标是找出邀请人员的最大价值总和,同时确保员工与直接领导不被同时邀请。文章详细介绍了使用两维度DP数组来记录选择或不选择当前节点时的最优解,并通过递归方式遍历整棵树来求得最终答案。

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

题意:

给定n个人,和他们的直系领导关系,每个人有一个值,

一个宴会邀请他们中的一些人,如果邀请了员工,他的直接领导就不能被邀请了,如果邀请了领导,他的直接员工就不能被邀请了

问邀请到的所有人的值最大是多少

思路:

给定的是一棵树的形状,不会存在环,由于在选某个人的时候选与不选,对他的领导和员工都有影响,所以需要动态的处理

需要用到dp,而每个点可选可不选,我们可以想到 再加一维表示,某个点选不选

需要注意的是:dp 用于最优解问题,最优子问题的解,我们往这一方面想法靠拢

所以:我们遇到每个结点的时候假设这个结点就是根结点,用 dp 【i】表示以这个点为根节点的树的最优值

但这样无法满足我们的需求,这个点对他上方的点有影响,所以用到了我们开的第二维: dp [ i ] [ 0 ] = 不选 i 时下方最优解,dp [ i ] [ 1 ]  = 选 i 时下方最优解

有了这样的递归过程后,每个结点对他的孩子结点可以决定怎么选,详见代码:


#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<set>
#include<queue>
#include<stack>
#include<map>
#define PI acos(-1.0)
#define in freopen("in.txt", "r", stdin)
#define out freopen("out.txt", "w", stdout)

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 6000 + 7, maxd = 20 + 7, mod = 1e9 + 7;
const int INF = 0x7f7f7f7f;

int n, a[maxn], vis[maxn];
int dp[maxn][2];
vector<int> vec[maxn];

void dfs(int id) {
    dp[id][1] = a[id];
    for(int i = 0; i < vec[id].size(); ++i) {
        int u = vec[id][i];
        dfs(u);
        dp[id][0] += max(dp[u][0], dp[u][1]);
        dp[id][1] += dp[u][0];
    }
}

int main() {

    while(~scanf("%d", &n) && n) {
        for(int i = 1; i <= n; ++i)
            scanf("%d", &a[i]);
        int a, b;
        memset(vis, 0, sizeof vis);
        memset(vec, 0, sizeof vec);
        memset(dp , 0, sizeof dp );
        while(~scanf("%d %d", &a, &b) && a+b) {
            vis[a] = true;
            vec[b].push_back(a);
        }
        int r;
        for(int i = 1; i <= n; ++i)
            if(!vis[i]) { r = i; break; }
        dfs(r);
        printf("%d\n", max(dp[r][0], dp[r][1]) );
    }

    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值