蓝桥杯java组 生命之树

题目描述

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

他给每棵树的每个节点（叶子也称为一个节点）上，都标了一个整数，代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集S，使得对于S中的任意两个点a,b，都存在一个点列 {a, v1, v2, …, vk, b} 使得这个点列中的每个点都是S里面的元素，且序列中相邻两个点间有一条边相连。

在这个前提下，上帝要使得S中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。

经过atm的努力，他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算，他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。

「输入格式」
第一行一个整数 n 表示这棵树有 n 个节点。
第二行 n 个整数，依次表示每个节点的评分。
接下来 n-1 行，每行 2 个整数 u, v，表示存在一条 u 到 v 的边。由于这是一棵树，所以是不存在环的。

「输出格式」
输出一行一个数，表示上帝给这棵树的分数。

「样例输入」
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5

「样例输出」
8
————————————————

代码

import java.util.*;  
  
public class Main {  
    public static void main(String[] args) {  
        System.out.println(new Solution().max);  
    }  
}  
class Solution {  
    ArrayList<List<Integer>> lists;  
    boolean[] isSet;  
    int[] weights;  
    long max = 0;  
  
    public Solution() {  
        Scanner scanner = new Scanner(System.in);  
        int n = scanner.nextInt();  
        weights = new int[n + 1];  
        for (int i = 1; i <= n; i++) {weights[i] = scanner.nextInt();}  
        lists = new ArrayList<>();  
        for (int i = 0; i <= n; i++) {  
            lists.add(new ArrayList<Integer>());  
        }  
        isSet = new boolean[n + 1];  
        for (int i = 0; i < n - 1; i++) {  
            int a = scanner.nextInt(), b = scanner.nextInt();  
            lists.get(a).add(b);  
            lists.get(b).add(a);  
        }  
        dfs(1);  
    }  
      
    public long dfs(int index){  
        isSet[index] = true;  
        long sum = weights[index];  
        for (Integer i : lists.get(index)) {  
            if (!isSet[i]){  
                long res = dfs(i);  
                if (res > 0) sum += res;  
            }  
        }  
        if (max < sum) max = sum;  
        return sum;  
    }  
}

题解

根据题目中对点集合S的描述很复杂, 其实简单理解就是一推连通的节点

而题目要求的就是符合条件的S的和的最大值
将S看作所有节点构成的树的子树, 我们无需关心这颗树长啥样, 只需关心它的和即可
如果简单求一整颗树的和很简单, 等于 所有子树和 + 该节点的值
那么在此基础上加入贪心的思想, 如果子树和小于0我就不把它算入和中
就可以在求各个子树的过程中保存到我们所需要的最大值

大致流程

以任意节点为根节点, 构造一棵树, 从根节点开始, 计算包含这个根节点的S的和的最大值
在各个节点最大值的计算过程中, 保存最大值即可.