派对的最大快乐值

题目描述
整个公司的人员结构可以看作是一棵标准的多叉树。树的头节点是公司唯一的老板，除老板外，每个员工都有唯一的直接上级，叶节点是没有任何下属的基层员工，除基层员工外，每个员工都有一个或多个直接下级，另外每个员工都有一个快乐值。
这个公司现在要办 party，你可以决定哪些员工来，哪些员工不来。但是要遵循如下的原则：
1.如果某个员工来了，那么这个员工的所有直接下级都不能来。
2.派对的整体快乐值是所有到场员工快乐值的累加。
3.你的目标是让派对的整体快乐值尽量大。
给定一棵多叉树，请输出派对的最大快乐值。
输入描述:
第一行两个整数 n 和 root，n 表示公司的总人数，root 表示公司的老板。

第二行 n 个整数 happy_i 表示员工 i 的快乐值。

接下来 n - 1 行每行两个整数 u_i 和 v_i 表示 u_i 是 v_i 的直接上级。
输出描述:
输出一个整数表示最大快乐值。
示例1
输入
3 1
5 1 1
1 2
1 3
输出
5

#include <bits/stdc++.h>
using namespace std;
struct Employee {
    int happy;
    std::vector<Employee*> subordinates;
    Employee(int h) : happy(h){}
};

struct Info {
    int yes;
    int no;
    Info(int y, int n) : yes(y), no(n){}
};

Info* process(Employee* x) {
    if(x->subordinates.empty()) {
        return new Info(x->happy, 0);
    }
    int yes = x->happy;
    int no = 0;
    for(Employee* next : x->subordinates) {
        Info *nextInfo = process(next);
        yes += nextInfo->no;
        no += std::max(nextInfo->yes, nextInfo->no);
    }
    return new Info(yes, no);
}

int maxHappy(Employee* boss) {
    if(boss == nullptr) {
        return 0;
    }
    Info *all = process(boss);
    return std::max(all->yes, all->no);
}

int main() {
    int n, root;
    cin >> n >> root;
    vector<Employee*> all_people;
    all_people.push_back({0});
    int happy_val;
    for(int i = 0; i < n; ++i) {
        cin >> happy_val;
        all_people.push_back(new Employee(happy_val));
    }
    int boss, sub;
    for(int i = 0; i < n - 1; i++) {
        cin >> boss >> sub;
        all_people[boss]->subordinates.push_back(all_people[sub]);
    }
    int res = maxHappy(all_people[root]);
    cout << res;
    return 0;
}

题目链接：https://www.nowcoder.com/practice/a5f542742fe24181b28f7d5b82e2e49a?tpId=101&tqId=33255&tPage=1&rp=1&ru=%2Fta%2Fprogrammer-code-interview-guide&qru=%2Fta%2Fprogrammer-code-interview-guide%2Fquestion-ranking&tab=answerKey