Java8 Lambda（分组和减少在一个单一的步骤）

Question

假设我有一个成对对象的列表，

List<Pair<A,B>> listOfPairs = //some list of pairs ; 

我想把这个列表分组成一张地图

目前，我可以分两步完成。第一步按A分组，并返回映射

   Map<A,Set<Pair<A,B>> intermediateStep =   listOfPairs.stream().collect(Collectors.groupingBy((Pair::getLeft), Collectors.toSet()));

然后，我对上面映射的入口集进行流式处理，并将它们收集到所需的最终结果中，主要是通过将每个Pair对象映射到其B值，然后将它们收集到一个集合：

 Map<A, Set<B>> finalResult= intermediateStep.entrySet().stream().collect(Collectors.toMap(e -> e.getKey(), e -> e.getValue().stream().map(Pair::getRight).collect(Collectors.toSet())));

有没有更好的方法，也许一步就能达到预期的效果？换句话说，上面提到的中间步骤按A分组，但分组的右侧返回整个Pair对象。我想按A分组，并在一个步骤中指向其关联的B集。

（我知道，仅仅因为我可以一步完成，并不意味着我应该这样做，因为可读性可能会受到影响，但我很好奇。）

谢谢！




Answer 1

您可以使用Map.computeIfAbent：

Map<A, Set<B>> finalResult = new HashMap<>();
listOfPairs.forEach(pair -> finalResult.computeIfAbsent(
        pair.getLeft(), 
        k -> new HashSet<>())
    .add(pair.getRight()));

如果要保留插入顺序，可以分别使用LinkedHashMap和LinkedHashSet而不是HashMap和HashSet。

Answer 2

您可以通过将映射收集器传递到分组By收集器来简化代码，如下所示：

Map<A, Set<B>> collect = 
       listOfPairs.stream()
                  .collect(Collectors.groupingBy(Pair::getLeft, 
                                Collectors.mapping(Pair::getRight,
                                       Collectors.toSet())));