题目描述:
在一棵树中,每个节点代表一个家庭成员,节点的数字表示其个人的财富值,一个节点及其直接相连的子节点被定义为一个小家庭。
现给你一棵树,请计算出最富裕的小家庭的财富和。
输入描述:
第一行为一个数N,表示成员总数,成员编号1-N,1<=N<=1000
第二行为N个空格分隔的数,表示编号1-N的成员的财富值。0<=财富值<=1000000
接下来N-1行,每行两个空格分隔的整数(N1N2),表示N1是N2的父节点。
输出描述:
最富裕的小家庭的财富和
示例1
输入:
4
100 200 300 500
1 2
1 3
2 4
输出:
700
说明:
成员1,2,3组成的小家庭财富值为600
成员2,4组成的小家庭财富值为700
示例2
输入:
4
100 200 300 500
1 2
1 3
1 4
输出:
1100
说明:
成员1,2,3,4组成的小家庭财富值为1100
JS参考解题:
class TreeNode {
constructor(val) {
this.val = val;
this.children = [];
}
}
function buildTree(N, wealth, edges) {
const nodes = new Array(N + 1);
for (let i = 1; i <= N; i++) {
nodes[i] = new TreeNode(wealth[i - 1]);
}
for (let [parent, child] of edges) {
nodes[parent].children.push(nodes[child]);
}
return nodes[1]; // 返回树的根节点
}
function maxWealth(root) {
let max = sumWealth(root);
for (let child of root.children) {
max = Math.max(max, sumWealth(child));
}
return max;
}
function sumWealth(node) {
let sum = node.val;
for (let child of node.children) {
sum += child.val;
}
return sum;
}
// 示例输入
const N = 4;
const wealth = [100, 200, 300, 500];
const edges = [[1, 2], [1, 3], [1, 4]];
// 构建树
const root = buildTree(N, wealth, edges);
// 计算最富裕的小家庭的财富和
const result = maxWealth(root);
console.log(result);