华为OD 寻找最富裕的小家庭（java）

题目描述：

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

输入描述：

第一行为一个数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

说明：

成员1,2,3组成的小家庭财富值为600

成员2,4组成的小家庭财富值为700

代码 

public class Main {
    public static void main(String[] args) {
        Scanner in=new Scanner(System.in);
        int n=in.nextInt();//家庭成员数
        int[] a=new int[n];//成员财富值
        for(int i=0;i<n;i++){
            a[i]=in.nextInt();
        }

        //父节点与财富值和
        Map<Integer,Integer> map=new HashMap<>();
        for(int i=0;i<(n-1);i++){
            int f= in.nextInt();
            int d=in.nextInt();
            //父节点不存在map中
            if(!map.containsKey(f)){
                map.put(f,a[d-1]+a[f-1]);
            }else{//父节点存在map中
                map.put(f,map.get(f)+a[d-1]);
            }
        }
        //遍历map找最大
        int max=0;
        Iterator<Map.Entry<Integer,Integer>> it=map.entrySet().iterator();
        while(it.hasNext()){
            Map.Entry<Integer,Integer> entry=it.next();
            max=Math.max(max,entry.getValue());
        }
        System.out.println(max);
    }
}