华为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);}
}

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