派对的最大快乐值

派对的最大快乐值

题目描述

整个公司的人员结构可以看作是一棵标准的多叉树。树的头节点是公司唯一的老板，除老板外，每个员工都有唯一的直接上级，叶节点是没有任何下属的基层员工，除基层员工外，每个员工都有一个或多个直接下级，另外每个员工都有一个快乐值。

这个公司现在要办 party，你可以决定哪些员工来，哪些员工不来。但是要遵循如下的原则：

1.如果某个员工来了，那么这个员工的所有直接下级都不能来。

2.派对的整体快乐值是所有到场员工快乐值的累加。

3.你的目标是让派对的整体快乐值尽量大。

给定一棵多叉树，请输出派对的最大快乐值。

输入描述：

第一行两个整数 n 和 root，n 表示公司的总人数，root 表示公司的老板。

第二行 n 个整数 $happ{y}_i$ 表示员工 i 的快乐值。

接下来 n - 1 行每行两个整数 $u_i$ 和 $v_i$ 表示 $u_i$ 是 $v_i$ 的直接上级。

输出描述：

输出一个整数表示最大快乐值。

示例1

输入

3 1
5 1 1
1 2
1 3

输出

5

备注：

$1\le n\le 500000$

$0\le happ{y}_i\le 1000$
输入保证是一棵树

---

题解：

树形 DP 模板题。

设 $f[u][0]$ 表示所有从以 u 为根的子树中选择，并且不选 u 这个点的方案中的最大快乐值

设 $f[u][1]$ 表示所有从以 u 为根的子树中选择，并且选择 u 这个点的方案中的最大快乐值

那么，则有：

$f[u][0]=\sum max(f[v][0],f[v][1])$

$f[u][1]=\sum f[v][0]$

v 是 u 的子节点。

最终的最大值就是：$max(f[root][0],f[root][1])$

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 500001;

int n, root;
int happy[N];
int e[N], h[N], ne[N], idx;
int f[N][2];

void add( int u, int v ) {
    e[++idx] = v;
    ne[idx] = h[u];
    h[u] = idx;
}

void dfs( int u ) {
    f[u][1] = happy[u];
    for ( int i = h[u]; i; i = ne[i] ) {
        int v = e[i];
        dfs( v );
        f[u][0] += max( f[v][0], f[v][1] );
        f[u][1] += f[v][0];
    }
}

int main(void) {
    scanf("%d%d", &n, &root);
    for ( int i = 1; i <= n; ++i )
        scanf("%d", happy + i);
    for ( int i = 1; i < n; ++i ) {
        int u, v;
        scanf("%d%d", &u, &v);
        add( u, v );
    }
    dfs( root );
    printf("%d\n", max( f[root][0], f[root][1] ));
    return 0;
}