😊😊 😊😊
不求点赞,只求耐心看完,指出您的疑惑和写的不好的地方,谢谢您。本人会及时更正感谢。希望看完后能帮助您理解算法的本质
😊😊 😊😊
题目描述:
在X森林里,上帝创建了生命之树。
他给每棵树的每个节点(叶子也称为一个节点)上,都标了一个整数,代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集 S,使得对于 S 中的任意两个点 a,b都存在一个点列 {a,v1,v2,…,vk,b} 使得这个点列中的每个点都是 S 里面的元素,且序列中相邻两个点间有一条边相连。
在这个前提下,上帝要使得 S 中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。
经过 atm 的努力,他已经知道了上帝给每棵树上每个节点上的整数。
但是由于 atm 不擅长计算,他不知道怎样有效的求评分。
他需要你为他写一个程序来计算一棵树的分数。
小白到进阶各种解法:
一、暴搜:待更新😊
代码:
二、记忆化搜索:待更新😊
代码:
三、本题考察算法:树形DP😊
思路:
-
首先明确答案是怎么来的?
一个序列,序列中的相邻的两个点是连接的 ⇒ 连通块。
所以说我们的答案来自于一棵树中的所有连通块中,和值最大的那个连通块。 -
暴力做法是:直接枚举所有连通块,即所有的答案,但这毫无疑问是会超时的。由于答案子集数量众多,所以我们试着去将答案进行一个分类!寻找分类的依据,观察所有的答案子集的特性,哪种性质可以作为划分集合的依据。
-
小经验:观察子集,从小到观察,从而确保无论小的子集还是大的子集,都能够被枚举到:而本题中的子集 – 连通块的种类,小到可以是一个叶子节点,一棵子树,大子树,整棵树 …;既然是树,那么必然有根节点啦,叶子节点是自己的根节点啦,所以我们可以通过根节点来将集合进行划分。
-
f [ u ] f[u] f[u] 表示以 u u u 为根的子树中所有包含 u u u 的连通块中权值的最大值。
-
那么如何去求连通块的最大值呢?如何求 f [ u ] f[u] f[u] 呢?
首先题目允许有负数的存在
如图所示:求以
u
u
u 为根的子树中所有连通块的最值。如果题目中没有负数的存在的话,那么对于
u
u
u 而言,肯定是直接将所有子树的权值进行累加,但是由于负数存在,而如果一个负数加入到连通块中,必然会使得连通块的总和变小,所以对于这种情况,我们还不如不选它,即取0;故可得
f
[
u
]
f[u]
f[u] 的计算公式:
f
[
u
]
=
w
[
u
]
+
∑
i
=
1
k
m
a
x
(
f
[
s
i
]
,
0
)
;
f[u] = w[u] + \sum\limits_{i=1}^kmax(f[s_i], 0);
f[u]=w[u]+i=1∑kmax(f[si],0);
所以说,我们应该先去递归求解子树,即叶子节点,然后回溯的时候更新计算!
代码:
#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e5 + 10;
typedef long long LL;
LL f[N];
int w[N];
int n;
int h[N], e[N*2], ne[N*2], idx;
void add (int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void dfs(int u, int father)
{
f[u] = w[u];
for (int i=h[u]; ~i; i = ne[i])
{
int j = e[i];
if (j == father) continue;
dfs(j, u);
f[u] += max(f[j], 0ll);
}
}
int main()
{
cin >> n;
for (int i=1; i <= n; i ++)
cin >> w[i];
memset(h, -1, sizeof(h));
for (int i=1; i < n; i ++)
{
int u, v;
cin >> u >> v;
add(u, v), add(v, u);
}
dfs(1, -1);
LL res=-1e8;
for (int i=1; i <= n; i ++)
res = max(res, f[i]);
cout << res << endl;
return 0;
}