题目描述
在 X X X森林里,上帝创建了生命之树。
他给每棵树的每个节点(叶子也称为一个节点)上,都标了一个整数,代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集
S
S
S,使得对于
S
S
S中的任意两个点
a
,
b
a,b
a,b,都存在一个点列
a
,
v
1
,
v
2
,
.
.
.
,
v
k
,
b
{a, v_1, v_2, ..., v_k, b}
a,v1,v2,...,vk,b 使得这个点列中的每个点都是
S
S
S里面的元素,且序列中相邻两个点间有一条边相连。
在这个前提下,上帝要使得
S
S
S中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。
经过 a t m atm atm的努力,他已经知道了上帝给每棵树上每个节点上的整数。但是由于 a t m atm atm 不擅长计算,他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。
输入格式
第一行一个整数
n
n
n 表示这棵树有
n
n
n 个节点。
第二行
n
n
n 个整数,依次表示每个节点的评分。
接下来
n
−
1
n-1
n−1 行,每行
2
2
2 个整数
u
,
v
u, v
u,v,表示存在一条 $u $到
v
v
v 的边。由于这是一棵树,所以是不存在环的。
输出格式
输出一行一个数,表示上帝给这棵树的分数。
样例输入
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
样例输出
8
数据范围
对于
30
%
30\%
30% 的数据,
n
<
=
10
n <= 10
n<=10
对于
100
%
100\%
100%的数据,
0
<
n
<
=
1
0
5
0 < n <= 10^5
0<n<=105, 每个节点的评分的绝对值不超过
1
0
6
10^6
106 。
资源约定
峰值内存消耗
<
256
M
< 256M
<256M
C
P
U
CPU
CPU消耗
<
3000
m
s
< 3000ms
<3000ms
思路
树形
D
P
DP
DP
给出的样例如图所示,红色部分圈出的是选取的最大值。
我们定义
f
[
u
]
f[u]
f[u],指以
u
u
u为顶点的评分最大的子树,包括
u
u
u本身。
上述是怎么求
f
[
u
]
f[u]
f[u],因为求最大的,所以我们只需要找到
f
[
u
]
f[u]
f[u]中评分大于等于0的子树,将他们加起来即可。所以
f
[
u
]
=
f
[
u
]
+
∑
i
=
1
k
m
a
x
(
0
,
f
[
i
]
)
f[u] = f[u] + \sum\limits_{i=1}^k max(0, f[i])
f[u]=f[u]+i=1∑kmax(0,f[i])
代码
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 100010, M = N * 2;
typedef long long LL;
int n;
int w[N];
int h[N], e[M], ne[M], idx;
LL f[N];
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void dfs(int u, int fa)
{
f[u] = w[u];
for ( int i = h[u]; ~i; i = ne[i] )
{
int j = e[i];
if ( j != fa )
{
dfs(j, u);
f[u] += max(0ll, f[j]);
}
}
}
int main()
{
cin >> n;
memset(h, -1, sizeof h);
for ( int i = 1; i <= n; i ++ ) cin >> w[i];
for ( int i = 0; i < n - 1; i ++ )
{
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
dfs(1, -1);
LL res = f[1];
for ( int i = 2; i <= n; i ++ ) res = max(res, f[i]);
cout << res << endl;
return 0;
}