华为od-C卷100分题目10 - 寻找最富裕的小家庭

华为od-C卷100分题目10 - 寻找最富裕的小家庭

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

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

public class Main {
    public static void main(String[] args) {
        int n;
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        int[] nums = new int[n];
        for (int i = 0; i < n; i++) {
            nums[i] = sc.nextInt();
        }
        int[] ans = Arrays.copyOf(nums, nums.length);
        int a, b;
        int max = Integer.MIN_VALUE;
        for (int i = 0; i < n - 1; i++) {
            a = sc.nextInt();
            b = sc.nextInt();
            ans[a - 1] += nums[b - 1];
            max = Math.max(ans[a - 1], max);
        }
        System.out.println(max);
    }
}

思路：两个数组，一个存原值，一个存总的财富值，根据结点序号找到对应财富值求和