没有上司的舞会

最新推荐文章于 2023-03-10 21:49:37 发布

原创最新推荐文章于 2023-03-10 21:49:37 发布 · 97 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++ #深度优先

算法基础学习专栏收录该内容

51 篇文章

订阅专栏

本文介绍树形dp经典例题“没有上司的舞会”。Ural大学职员关系呈树状，每个职员有快乐指数，职员不愿与直接上司参会，主办方要在满足条件下使参会职员快乐指数总和最大。文中给出输入输出格式及数据范围。

没有上司的舞会

树形dp经典例题
分开讨论一下就行了

Ural 大学有 N 名职员，编号为 1∼N。

他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。

每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。

现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。

在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式
第一行一个整数 N。

接下来 N 行，第 i 行表示 i 号职员的快乐指数 Hi。

接下来 N−1 行，每行输入一对整数 L,K，表示 K 是 L 的直接上司。

输出格式
输出最大的快乐指数。

数据范围
1≤N≤6000,
−128≤Hi≤127
输入样例：

输出样例：

#include <iostream>
#include <vector>

using namespace std;
const int N = 6e3+10;
vector<int> tu[N];
int f[N][2];
int n;
int happy[N];
bool father[N];

void dfs(int u){
    f[u][1]+=happy[u];
    for(auto i:tu[u]){
        dfs(i);
        f[u][0] += max(f[i][0],f[i][1]);
        f[u][1] += f[i][0];
    }
}

int main() {
    cin>>n;
    for(int i = 1;i<=n;i++)
        scanf("%d",&happy[i]);
    for(int i = 1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        tu[b].push_back(a);
        father[a] = true;
    }

    int root = 1;
    while(father[root]) root++;

    dfs(root);

    cout<<max(f[root][0],f[root][1]);

    return 0;
}