SortedMap，其中键是一个属性，值是具有相同键的n个最低值的降序列表

Question

我正在使用一个CSV文件，其中包含一些有关事故的信息。现在我已经创建了事故类型：

private Integer driverAge;
private Integer vehicleAge;

public Accident(Integer driverAge, Integer vehicleAge) {
this.driverAge = driverAge;
this.vehicleAge = vehicleAge;
}

我还创建了一个函数，可以读取所有csv文件，将所有事故转换为列表

private List<Accident> accidents;

public AccidentArchive(List<Accident> accidents) {
this.accidents = accidents;
}

因此，我们正在处理我还不完全理解的流，我一直在做这个练习，我必须创建一个返回SortedMap的函数

public SortedMap<Integer, List<Integer>> getNMinVehicleAgesPerDriverAge(Integer n) {
return getAccidents().stream().
...

我试过使用收集器。toMap（）和收集器。toList（）以某种方式使其工作，但我不知道如何做。我已经搜索了一段时间，但我没有找到任何与我的情况类似的东西，在任何文档页面中也找不到任何有用的东西。

Answer 1

您可以使用groupingBy（）的风格，它需要三个参数：分类器函数、供应商mapFactory（允许指定生成的map类型）和下游收集器。

作为groupingBy（）的下游收集器，您可以使用映射（）和自定义收集器的组合，它们只保留与每个驾驶率相关的最低车辆里程值：

public SortedMap<Integer, List<Integer>> getNMinVehicleAgesPerDriverAge(Integer n) {
    return getAccidents().stream()
        .collect(Collectors.groupingBy(Accident::getDriverAge,
            TreeMap::new,
            Collectors.mapping(Accident::getVehicleAge, 
                getMaxN(n, Comparator.<Integer>reverseOrder()))));
}

下面提供的方法负责根据所提供的结果列表的最大大小和比较器生成自定义收集器。答案中详细解释了其背后的逻辑：

public static <T> Collector<T, ?, List<T>> getMaxN(int size, Comparator<T> comparator) {
        
    return Collector.of(
        () -> new PriorityQueue<>(comparator),
        (Queue<T> queue, T next) -> tryAdd(queue, next, comparator, size),
        (Queue<T> left, Queue<T> right) -> {
            right.forEach(next -> tryAdd(left, next, comparator, size));
            return left;
        },
        (Queue<T> queue) -> queue.stream().toList(),
        Collector.Characteristics.UNORDERED);
}
    
public static <T> void tryAdd(Queue<T> queue, T next, Comparator<T> comparator, int size) {
    if (queue.size() == size && comparator.compare(next, queue.element()) < 0) queue.remove(); // if next value is less than the smallest element in the queue and max size has been exceeded the largest element needs to be removed from the queue
    if (queue.size() < size) queue.add(next);
}

顺便说一句，如果您的作业没有指定使用SortedMap作为返回类型的要求。最好使用NavigableMap接口，它定义了更广泛的方法。