游乐园的游览计划

/*
 * 思路：
 * 1. 使用Stream API来构建图并处理数据。
 * 2. 遍历每个项目A，查找符合条件的路径并计算喜爱值。
 * 3. 使用Stream的map和filter方法来筛选和计算数据。
 */

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

public class MaxHappinessStream {
    public int maxHappiness(int[][] edges, int[] value) {
        int n = value.length;
        Map<Integer, List<Integer>> graph = IntStream.range(0, n)
                .boxed()
                .collect(Collectors.toMap(i -> i, i -> new ArrayList<>()));

        Arrays.stream(edges).forEach(edge -> {
            graph.get(edge[0]).add(edge[1]);
            graph.get(edge[1]).add(edge[0]);
        });

        return IntStream.range(0, n).map(A -> graph.get(A).stream()
                .flatMap(B -> graph.get(B).stream()
                        .filter(C -> C != A && graph.get(C).contains(A))
                        .map(C -> new int[]{A, B, C}))
                .mapToInt(abc -> {
                    int A1 = abc[0], B1 = abc[1], C1 = abc[2];
                    Set<Integer> visited = new HashSet<>(Arrays.asList(A1, B1, C1));
                    return graph.get(A1).stream()
                            .flatMap(B2 -> graph.get(B2).stream()
                                    .filter(C2 -> !visited.contains(B2) && !visited.contains(C2) && graph.get(C2).contains(A1))
                                    .mapToInt(C2 -> value[A1] + value[B1] + value[C1] + value[B2] + value[C2])
                            ).max().orElse(0);
                }).max().orElse(0)).max().orElse(0);
    }
}