LDUOJ——没有上司的晚会(树形DP)

简介: LDUOJ——没有上司的晚会(树形DP)

原题链接

题意:

给一棵树,每个点都有权值a[i],要求从中选若干个点,满足该点和其父亲节点不能同时被选择,求最大权值。

思路:

一般树形DP的第一维都是子树的编号,我们用dp[i]表示选完以i为根节点的子树能够获得的最大权值。

一个点是否被选择,和他的父节点是否被选择有关,在此基础上再加一维度表示该点是否被选择,即dp[i] [0]表示不选择节点i且选完以i为根节点的子树能够获得的最大权值,dp[i] [1]表示选择节点i且选完以i为根节点的子树能够获得的最大权值。

再来考虑状态转移,因为子节点和父节点不能被同时选择,所以当选择父节点时,就不能再选择直接的子节点,即dp[i][1]=a[i]+max(dp[j][0]) j∈i的子树;当不选择父节点时,是否选择子节点都可以,即dp[i][0]=max(dp[j][0],dp[j][1]),j∈i的子树;

答案即max(dp[root][0],dp[root][1]);

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=6000+100;
vector<int>v[maxn];
bool st[maxn];///是否没有上司
int a[maxn];///快乐指数
int n;
int dp[maxn][2];
void dfs(int u){
    dp[u][1]=a[u];
    for(auto t:v[u]){
        dfs(t);
        dp[u][1]+=dp[t][0];
        dp[u][0]+=max(dp[t][0],dp[t][1]);
    }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<n;i++){
        int l,k;
        cin>>l>>k;
        ///k是l的直接上司 k指向l
        st[l]=1;
        v[k].push_back(l);
    }
    int root=1;
    while(st[root]) root++;
    dfs(root);
    cout<<(max(dp[root][0],dp[root][1]));
    return 0;
}

参考:《算法竞赛进阶指南》


目录
相关文章
|
9月前
|
测试技术
蓝桥 晚会节目单 (线段树)
蓝桥 晚会节目单 (线段树)
|
9月前
|
算法
【过河卒】回溯算法保姆式解题
【过河卒】回溯算法保姆式解题
53 0
|
9月前
|
算法 C语言 C++
【数论】蚂蚁感冒、饮料换购、买不到的数目
长 100 厘米的细长直杆子上有 n只蚂蚁。
56 0
|
10月前
2022年9月月赛乙组 T3.棋盘问题
2022年9月月赛乙组 T3.棋盘问题
|
11月前
|
算法 测试技术
蓝桥杯2022年第十三届决赛真题-卡牌——二分法
蓝桥杯2022年第十三届决赛真题-卡牌——二分法
88 0
|
11月前
|
测试技术
蓝桥杯2021年第十二届省赛真题-砝码称重(动态规划)
蓝桥杯2021年第十二届省赛真题-砝码称重(动态规划)
三道好题分享
上课睡觉 - AcWing题库
61 0
|
Android开发
LeetCode 双周赛 98,脑筋急转弯转不过来!
大家好,我是小彭。 昨晚是 LeetCode 第 98 场双周赛,你参加了吗?这场周赛需要脑筋急转弯,转不过来 Medium 就会变成 Hard,转得过来就变成 Easy。
68 0
洛谷P2016 战略游戏 (树形dp)
洛谷P2016 战略游戏 (树形dp)
117 0