华为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);}
}
思路:两个数组,一个存原值,一个存总的财富值,根据结点序号找到对应财富值求和