lambda - 使用 Java 8 Stream API 展平地图并关联值

Question

假设我有一组字符串到整数值的映射：
Map<HashSet<String>, Integer> map = new HashMap<>().

例如，mapis（我们假设没有重复的字符串）：

{x,y}   ->  2
{z}     ->  3
{u,v,w} ->  4

如何使用Java 8 Stream API获得如下another_map类型：Map<String, Integer>

x -> 2
y -> 2
z -> 3
u -> 4
v -> 4
w -> 4

它看起来像一个flatMap操作，但是我怎样才能将整数值与每个字符串键适当地关联起来呢？

Answer 1

您可以Map.Entry像这样创建中间对象：

Map<String, Integer> result = map.entrySet().stream()
   .<Entry<String, Integer>>flatMap(entry -> 
       entry.getKey()
            .stream()
            .map(s -> new AbstractMap.SimpleImmutableEntry<>(s, entry.getValue())))
   .collect(Collectors.toMap(Entry::getKey, Entry::getValue));

或者，您可以使用项目中的任何其他对/元组类型。

请注意，我的免费StreamEx库支持以更简洁的方式处理此类情况（内部与上述相同）：

Map<String, Integer> result = EntryStream.of(map).flatMapKeys(Set::stream).toMap();

该类EntryStream扩展Stream<Map.Entry>并提供了其他有用的方法，例如flatMapKeysor toMap。