java Stream倒序收集到TreeMap

Question

我有以下代码从流中收集 map ，并将其转换为树形图

Map<String, String> result = myMap.values().stream().collect(Collectors.toMap(Person::getScore, Person::getName));
    return new TreeMap<>(result);

有没有办法在同一行中返回降序排序的 treeMap？我不能将比较器作为参数传递。

Answer 1

我想你可以使用四个参数 toMap返回收集器的方法。假设 getScore 方法返回一个 Integer(或者您可以继续使用 String，但我认为它不太可用):

//supplier of TreeMap with your custom comparator
Supplier<TreeMap<Integer, String>> myMapSupplier = () -> new TreeMap<>(Comparator.reverseOrder());

Map<Integer, String> result = myMap.values()
             .stream()
             .collect(Collectors.toMap(Person::getScore, Person::getName, (s, s2) -> s, myMapSupplier));

最后一个元素是Map 的供应商。作为documentation状态:

mapSupplier - a function which returns a new, empty Map into which the results will be inserted

因此您仍然可以在一次调用中拥有一个带有自定义 Comparator 的 TreeMap。