如何变换Map,B>成 Map>

Question

我在寻找最佳解决方案时遇到问题（这不是多次使用 forEach）。

我有对象 A、B 和 C。对象B包含一组对象C（即B->Set`）。

我收到了 Map, B>，我需要将其转换为 Map>。

流库中是否有任何函数可以提供这种特定的解决方案？

我现在想出的解决方案是通过forEach来处理这个地图。

在每次迭代中，我都会合并地图 Map> 中每个 A 的 C 集。这是解决方案的概要（里面发生了很多事情，b->c是发生的事情的快捷方式）。

通过创建 Map>，在每次迭代中我为每个迭代提供先前集合的合并。这是解决方案的概要（里面发生了很多事情，b->c是正在发生的事情的快捷方式）。

Map<A, Set<C>> aSetCMap = new HashMap<>();
for (Map.Entry<Set<A>, B> entry : setABMap.EntrySet()) {
  var setA = entry.getKey();
  var setC = b.transformC();
  setA.forEach(item -> aSetCMap.merge(item, setC,
    (current, added) -> Streams.concat(current.stream(), added.stream())
      .collect(Collectors.toSet())));
}

我想出的解决方案是一种在 O(n^3) 时间内完成的奇怪方法。我正在使用它的东西需要高性能，因此我的问题。

Answer 1

您需要展平每个条目Map.Entry, B>。并将每个对象A与从对象B提取的Set相关联。

之后，通过使用 Collectors.toMap() 风格来应用 collect()，它采用三个参数：一个 keyMapper、一个 valueMapper 和一个 mergeFunction （需要组合映射到同一对象 A 的集合）

Map<Set<A>, B> source = getSourceMap();

Map<A, Set<C>> result =
    source.entrySet().stream()
        .flatMap(entry -> entry.getKey().stream()
                    .map(a -> Map.entry(a,  entry.getValue().getC())))
        .collect(Collectors.toMap(Map.Entry::getKey, 
                                  Map.Entry::getValue,
                                  (set1, set2) -> {set1.addAll(set2); return set1;}));

上面显示的方法适用于以下情况这Set可以在恒定时间内从对象B获得。在这里我依赖你的话：

B 包含一组对象C

如果此操作需要额外的处理（我在代码中看到一个方法 transformC()）并创建一个新的每次调用 transformC() 后设置，然后对于源映射中的每个条目，必须仅执行一次（即使用多行 lambda ）。

对于基于流的解决方案的时间复杂度，您对性能改进的期望将不会实现（假设迭代解决方案中没有逻辑错误）。

注意，使用可变对象 (Set) 作为键的情况本身就是恶性的，可能会导致不正确的行为。