使用流和 lambda 将 Map<Integer, List<String>> 展平为 Map<String, Integer>

Question

我想展平一个Map，它将Integer 键与String 列表相关联，而不会丢失键映射。 我很好奇，好像使用 stream 和 lambda 这样做是可能和有用的。

我们从这样的开始：

Map<Integer, List<String>> mapFrom = new HashMap<>();

让我们假设 mapFrom 填充在某个地方，看起来像：

1: a,b,c
2: d,e,f
etc.

我们还假设列表中的值是唯一的。

现在，我想“展开”它以获得第二张地图，例如：

a: 1
b: 1
c: 1
d: 2
e: 2
f: 2
etc.

我可以这样做（或者非常类似，使用foreach）：

Map<String, Integer> mapTo = new HashMap<>();
for (Map.Entry<Integer, List<String>> entry: mapFrom.entrySet()) {
    for (String s: entry.getValue()) {
        mapTo.put(s, entry.getKey());
    }
}

现在让我们假设我想使用 lambda 而不是嵌套的 for 循环。我可能会这样做：

Map<String, Integer> mapTo = mapFrom.entrySet().stream().map(e -> {
    e.getValue().stream().?
    // Here I can iterate on each List, 
    // but my best try would only give me a flat map for each key, 
    // that I wouldn't know how to flatten.
}).collect(Collectors.toMap(/*A String value*/,/*An Integer key*/))

我也尝试了flatMap，但我认为这不是正确的方法，因为虽然它帮助我摆脱了维度问题，但我在这个过程中失去了关键。

简而言之，我的两个问题是：

• 是否可以使用streams 和lambda 来实现这一点？
• 这样做是否有用（性能、可读性）？

Answer 1

与 Java 9 的先前答案相同：

Map<String, Integer> mapTo = mapFrom.entrySet()
        .stream()
        .flatMap(entry -> entry.getValue()
                .stream()
                .map(s -> Map.entry(s, entry.getKey())))
        .collect(toMap(Entry::getKey, Entry::getValue));

Answer 2

您应该使用flatMap，如下所示：

entrySet.stream()
        .flatMap(e -> e.getValue().stream()
                       .map(s -> new SimpleImmutableEntry(e.getKey(), s)));

SimpleImmutableEntry 是AbstractMap 中的一个嵌套类。

Answer 3

您需要使用flatMap 将值展平到一个新流中，但是由于您仍然需要将原始键收集到Map 中，因此您必须映射到一个保存键和值的临时对象，例如

Map<String, Integer> mapTo = mapFrom.entrySet().stream()
       .flatMap(e->e.getValue().stream()
                    .map(v->new AbstractMap.SimpleImmutableEntry<>(e.getKey(), v)))
       .collect(Collectors.toMap(Map.Entry::getValue, Map.Entry::getKey));

Map.Entry 是不存在的元组类型的替代，任何其他能够容纳两个不同类型对象的类型就足够了。

另一种不需要这些临时对象的方法是自定义收集器：

Map<String, Integer> mapTo = mapFrom.entrySet().stream().collect(
    HashMap::new, (m,e)->e.getValue().forEach(v->m.put(v, e.getKey())), Map::putAll);

这与toMap 的不同之处在于以静默方式覆盖重复键，而没有合并功能的toMap 如果存在重复键，则会抛出异常。基本上，这个自定义收集器是一个具有并行能力的变体

Map<String, Integer> mapTo = new HashMap<>();
mapFrom.forEach((k, l) -> l.forEach(v -> mapTo.put(v, k)));

但请注意，即使输入图非常大，此任务也不会从并行处理中受益。只有当流管道中有额外的计算密集型任务可以从 SMP 中受益时，才有机会从并行流中受益。因此，简洁、顺序的 Collection API 解决方案或许更可取。

Answer 4

希望这会以最简单的方式完成。 :))

mapFrom.forEach((key, values) -> values.forEach(value -> mapTo.put(value, key)));

Answer 5

这应该有效。请注意，您丢失了 List 中的一些密钥。

Map<Integer, List<String>> mapFrom = new HashMap<>();
Map<String, Integer> mapTo = mapFrom.entrySet().stream()
        .flatMap(integerListEntry -> integerListEntry.getValue()
                .stream()
                .map(listItem -> new AbstractMap.SimpleEntry<>(listItem, integerListEntry.getKey())))
        .collect(Collectors.toMap(AbstractMap.SimpleEntry::getKey, AbstractMap.SimpleEntry::getValue));