2024年华为OD机试真题-最富裕的小家庭-(C++/Java/python)-OD统一考试（C卷D卷）

题目描述

在一颗树中，每个节点代表一个家庭成员，节点的数字表示其个人的财富值，

一个节点及其直接相连的子节点被定义为一个小家庭。

现给你一颗树，请计算出最富裕的小家庭的财富和。

输入描述

第一行为一个数 N，表示成员总数，成员编号 1~N。1 ≤ N ≤ 1000

第二行为 N 个空格分隔的数，表示编号 1~N 的成员的财富值。0 ≤ 财富值 ≤ 1000000

接下来 N -1 行，每行两个空格分隔的整数（N1, N2），表示 N1 是 N2 的父节点。

输出描述

最富裕的小家庭的财富和

用例1

输入

4

100 200 300 500

1 2

1 3

2 4

输出

700

说明

一个节点及其直接相连的子节点财富和最大的为节点2和节点

4的组合，财富和为200加500为700.

考点

逻辑分析

代码 

python

n=int(input())
 
val=list(map(int,input().split()))
 
arr=[]
 
for i in range(n-1):
    a,b=list(map(int,input().split()))
    arr.append([a-1,b-1])
 
res=0
 
for i in arr:
    i1,i2=i
    res=max(res,val[i1]+val[i2])
 
print(res)

java

import java.util.Arrays;
import java.util.Scanner;
 
public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
 
    int n = sc.nextInt();
 
    long[] wealth = new long[n + 1];
    long[] family = new long[n + 1];
    for (int i = 1; i <= n; i++) {
      wealth[i] = sc.nextInt();
      family[i] = wealth[i];
    }
 
    for (int i = 0; i < n - 1; i+