假设我的原始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 Pre 8),但是如何使用Java Stream API实现同样的方法?
Map<String, Set<String>> reversed = original.entrySet().stream().map(x -> ????).collect(??)
这里有类似的问题,但这仅适用于单个有价值的Map
s。
您可以使用flatMap
将Map
分解为键值对(每个键和值为单个String
),然后根据需要收集它们:
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]}
更急切但更简单的解决方案是使用 forEach
:
Map<String, Set<String>> original,result; // initialised
original.forEach((key, value) -> value.forEach(v ->
result.computeIfAbsent(v, k -> new HashSet<>()).add(key)));