从列表对象创建一个sortedmap，其值表示为n个最低对象的列表列表的列表，映射到特定键

Question

我正在使用一个CSV文件，其中包括有关事故的一些信息。

我已经创建了事故类型：

private Integer driverAge;
private Integer vehicleAge;

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

我还创建了一个读取所有CSV文件的函数，将所有事故转换为list＆lt; atest＆gt;并将其保存到此类型意外结构：

private List<Accident> accidents;

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

因此，我们正在使用我尚不完全理解的流，​​并且我一直在练习中，我必须制作返回sortedmap＆lt的函数; k，v＆gt;，其中键必须是drigrage值，并且值必须为 list 在 n最低车辆车辆 值值：

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

我尝试使用collector.tomap（） and collector.tolist（）以某种方式使它起作用，但我不知道该怎么做。

Answer 1

简化方法

此问题与通过部分排序查找n最大值（或最小）值的算法问题相关。它是使用收藏家的实施似乎很难的，因此我决定引入简化的解决方案。

我们可以使用：

• A classifier function ，
• a 供应商 mapfactory（允许指定结果类型地图）
• 和下游收集器。

作为groupingby（）的下游收集器，我们可以使用collectors mapping> mapping（） and code> and code> tolist（）代码>和a 函数将对映射到每个 key 的整个结果列表进行排序，然后将删除仅保留n最低Vehicleage ：

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

我之前说过的更多性能版本

，我们不需要对映射到每个键的所有值进行排序。与列表的总尺寸相比，n是矮人（例如3 to 100,000每个键），这将导致严重的性能命中。

我们可以使用PriorityQueue（这是JDK中内置的HEAP数据结构的实现）来引入部分分类。

为了增强先前的解决方案，我们需要替换groupingby（）的下游收集器，您可以使用mapping（）和自定义收集器由Priorityqueue支持，该>仅保留n最低vericleage与每个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()))));
}

：负责根据所得列表的提供的最大大小和A 比较器来生成自定义收集器。

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接口，该接口定义了更广泛的方法。