从列表对象创建一个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;
}

因此，我们正在处理我还不完全理解的流，我一直在做这个练习，我必须生成一个返回排序映射的函数

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

我尝试使用Collectors.toMap（）和Collectors.toList（）以某种方式使其工作，但我不知道如何做到这一点。

Answer 1

您可以使用一种风格的groupingBy（），它需要三个参数：分类器函数、供应商映射工厂（允许指定映射的结果类型）和下游收集器。

作为groupingBy（）的下游收集器，您可以利用映射（）和仅保留n最低车辆年龄值的自定义收集器的组合与每个driverage相关联：

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);
}

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