收集树上所有苹果的最少时间

/*
 * 思路：
 * 1. 构建树的表示，使用邻接表。
 * 2. 使用深度优先搜索（DFS）来遍历树。
 * 3. 对于每个节点，如果其子树中有苹果，计算需要的时间并返回。
 * 4. 如果一个节点或其子节点有苹果，增加相应的时间。
 * 5. 使用Java Stream API来构建树和进行DFS遍历。
 */

import java.util.*;
import java.util.stream.*;

public class SolutionStream {
    public int minTime(int n, int[][] edges, List<Boolean> hasApple) {
        // 构建邻接表
        Map<Integer, List<Integer>> tree = Arrays.stream(edges)
                .flatMap(edge -> Stream.of(Map.entry(edge[0], edge[1]), Map.entry(edge[1], edge[0])))
                .collect(Collectors.groupingBy(Map.Entry::getKey, Collectors.mapping(Map.Entry::getValue, Collectors.toList())));
        // 记录访问节点避免重复
        boolean[] visited = new boolean[n];
        return dfs(0, tree, hasApple, visited);
    }

    private int dfs(int node, Map<Integer, List<Integer>> tree, List<Boolean> hasApple, boolean[] visited) {
        visited[node] = true;
        int totalTime = tree.getOrDefault(node, Collections.emptyList()).stream()
                .filter(child -> !visited[child])
                .mapToInt(child -> {
                    int timeSpent = dfs(child, tree, hasApple, visited);
                    return (timeSpent > 0 || hasApple.get(child)) ? timeSpent + 2 : 0;
                })
                .sum();
        return totalTime;
    }
}