如何让Java Stream的max操作达到指定阈值时提前终止并包含临界值？

问题核心 #

你当前的问题在于takeWhile的行为是谓词返回false时丢弃当前元素并截断流，而你需要的是「保留第一个命中上限的元素，再截断流」，同时保留求最大值的逻辑。
JDK 原生 Stream 没有内置的takeWhileInclusive这类包含触发元素的截断操作，但可以通过以下两种内置API实现的方案满足需求：

方案1：有状态谓词实现（仅适用串行流）

通过原子标记记录是否首次命中上限，调整takeWhile的判断逻辑，确保命中上限的当前元素被保留后再截断流：

public Solution best() {
    AtomicBoolean foundUpperLimit = new AtomicBoolean(false);
    int maxSize = problem.animals().size();
    return StreamSupport.stream(this, false) // 必须串行，并行场景下状态会有并发问题
        .takeWhile(sol -> {
            // 已经找到过上限元素，直接截断后续元素
            if (foundUpperLimit.get()) return false;
            // 当前元素命中上限，标记后保留当前元素再截断后续
            if (sol.pairs().size() >= maxSize) {
                foundUpperLimit.set(true);
            }
            // 无论是否命中上限，当前元素都保留
            return true;
        })
        .max(Solution.byMatched)
        .orElseThrow();
}

方案2：异常短路实现（支持并行流，性能最优）

因为你的比较器维度就是pairs.size()，只要出现size等于上限的元素，它必然是全局最大值，不需要继续遍历。可以通过抛自定义无栈异常的方式直接短路整个流处理，并行场景下也适用：

public Solution best() {
    int maxSize = problem.animals().size();
    try {
        return StreamSupport.stream(this, true)
            .peek(sol -> {
                if (sol.pairs().size() == maxSize) {
                    throw new FoundMaxException(sol);
                }
            })
            .max(Solution.byMatched)
            .orElseThrow();
    } catch (FoundMaxException e) {
        return e.getSolution();
    }
}

// 自定义无栈异常，避免栈追踪的性能开销
private static class FoundMaxException extends RuntimeException {
    private final Solution solution;
    public FoundMaxException(Solution solution) {
        super(null, null, false, false);
        this.solution = solution;
    }
    public Solution getSolution() {
        return solution;
    }
}

补充说明

如果你的比较器后续可能调整，不止比较pairs.size，方案1的逻辑仍然适用，方案2则需要调整短路判断逻辑。

内容的提问来源于stack exchange，提问作者Reinderien