【华为OD机试真题】- 寻找最富裕的小家庭（JS解答）

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

输入描述:
第一行为一个数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);