Lang:Java
Edit12345678910111213141516171819202122232425262728293031import java.util.ArrayList;import java.util.Arrays;import java.util.List;import java.util.Map;import java.util.Scanner;import java.util.stream.Collectors;class Main {static class Node {int val;List<Node> children = new ArrayList<>();Node parent;Node(int v) {val = v;}}public static void main(String[] args) {long res = 0;Scanner in = new Scanner(System.in);int N = in.nextInt();Node[] arr = new Node[N];for (int i = 0; i < N; i++) {arr[i] = new Node(in.nextInt());}for (int i = 1; i < N; i++) {int a = in.nextInt() - 1;int b = in.nextInt() - 1;arr[a].children.add(arr[b]);arr[b].parent = arr[a];}