生命之树【蓝桥杯真题：暴搜 + DP】

😊😊 😊😊
不求点赞，只求耐心看完，指出您的疑惑和写的不好的地方，谢谢您。本人会及时更正感谢。希望看完后能帮助您理解算法的本质
😊😊 😊😊

题目描述：

在X森林里，上帝创建了生命之树。

他给每棵树的每个节点（叶子也称为一个节点）上，都标了一个整数，代表这个点的和谐值。

上帝要在这棵树内选出一个非空节点集 S，使得对于 S 中的任意两个点 a,b都存在一个点列 {a,v1,v2,…,vk,b} 使得这个点列中的每个点都是 S 里面的元素，且序列中相邻两个点间有一条边相连。

在这个前提下，上帝要使得 S 中的点所对应的整数的和尽量大。

这个最大的和就是上帝给生命之树的评分。

经过 atm 的努力，他已经知道了上帝给每棵树上每个节点上的整数。

但是由于 atm 不擅长计算，他不知道怎样有效的求评分。

他需要你为他写一个程序来计算一棵树的分数。

小白到进阶各种解法：

一、暴搜：待更新😊

代码：

二、记忆化搜索：待更新😊

代码：

三、本题考察算法：树形DP😊

思路：

1. 首先明确答案是怎么来的？
   
   一个序列，序列中的相邻的两个点是连接的 ⇒ 连通块。
   所以说我们的答案来自于一棵树中的所有连通块中，和值最大的那个连通块。

2. 暴力做法是：直接枚举所有连通块，即所有的答案，但这毫无疑问是会超时的。由于答案子集数量众多，所以我们试着去将答案进行一个分类！寻找分类的依据，观察所有的答案子集的特性，哪种性质可以作为划分集合的依据。

3. 小经验：观察子集，从小到观察，从而确保无论小的子集还是大的子集，都能够被枚举到：而本题中的子集 – 连通块的种类，小到可以是一个叶子节点，一棵子树，大子树，整棵树 …；既然是树，那么必然有根节点啦，叶子节点是自己的根节点啦，所以我们可以通过根节点来将集合进行划分。
   

4. $f[u]$ 表示以 $u$ 为根的子树中所有包含 $u$ 的连通块中权值的最大值。

5. 那么如何去求连通块的最大值呢？如何求 $f[u]$ 呢？
   首先题目允许有负数的存在
   

如图所示：求以 $u$ 为根的子树中所有连通块的最值。如果题目中没有负数的存在的话，那么对于 $u$ 而言，肯定是直接将所有子树的权值进行累加，但是由于负数存在，而如果一个负数加入到连通块中，必然会使得连通块的总和变小，所以对于这种情况，我们还不如不选它，即取0；故可得 $f[u]$ 的计算公式：
$f[u]=w[u]+\sum _{i=1}^{k}max(f[s_i],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;
}