华为OD机试 Python 【最富裕的小家庭】

题目

在一颗树中，每个节点代表一个家庭成员，节点的数字表示其个人的财富值，一个节点及其直接相连的子节点被定义为一个小家庭。
现给你一颗树，请计算出最富裕的小家庭的财富和。

输入描述
第一行为一个数 N，表示成员总数，成员编号 1~N。1 ≤ N ≤ 1000
第二行为 N 个空格分隔的数，表示编号 1~N 的成员的财富值。0 ≤ 财富值 ≤ 1000000
接下来 N -1 行，每行两个空格分隔的整数（N1, N2），表示 N1 是 N2 的父节点。

输出描述
最富裕的小家庭的财富和

用例
输入
4
100 200 300 500
1 2
1 3
2 4
输出
700

解题思路

处理输入：
首先，我们将家庭成员的财富存储在一个列表中。
为了让财富列表的索引与成员编号对应起来，我们在财富列表的开头插入一个零。
然后，我们创建一个家庭成员列表，初始时与财富列表相同。
更新家庭财富：
遍历每个父母-子女关系，根据关系更新家庭成员的财富。
对于每对父母-子女关系，我们将子女的财富加到父母的财富中。
找出最大财富：
在所有家庭成员中找到最大的财富值即为最大家庭财富。

方法的时间复杂度为 O(n)，其中 n 是家庭成员的数量。因为我们只需要一次遍历来更新家庭财富和一次遍历来找出最大财富。

代码