首页 > 解决方案 > 当原始地图包含集合作为值时如何创建反向地图?

问题描述

假设我的原件Map包含以下内容:

Map<String, Set<String>> original = Maps.newHashMap();
original.put("Scott", Sets.newHashSet("Apple", "Pear", "Banana");
original.put("Jack", Sets.newHashSet("Banana", "Apple", "Orange");

我想创建一个Map包含以下内容的反转:

  "Apple":  ["Scott", "Jack"]
  "Pear":   ["Scott"]
  "Banana": ["Scott", "Jack"]
  "Orange": ["Jack"]

我知道它可以以旧方式(Java 8 之前)完成,但是如何使用 Java Stream API 实现相同的功能?

Map<String, Set<String>> reversed = original.entrySet().stream().map(x -> ????).collect(??)

这里发布了类似的问题,但这仅适用于单值Maps。

标签: javajava-8java-stream

解决方案


您可以使用 将其分解Map为键值对(其中每个键和值都是一个StringflatMap,然后根据需要收集它们:

Map<String,Set<String>> rev =
    original.entrySet ()
            .stream ()
            .flatMap (e -> e.getValue ()
                            .stream ()
                            .map (v -> new SimpleEntry<String,String>(v,e.getKey ())))
            .collect(Collectors.groupingBy (Map.Entry::getKey,
                                            Collectors.mapping (Map.Entry::getValue, 
                                                                Collectors.toSet())));
System.out.println (rev);

输出:

{Apple=[Jack, Scott], Pear=[Scott], Orange=[Jack], Banana=[Jack, Scott]}

推荐阅读