从 Map<String, Set<Integer>> 从 Set<String> 的键中收集值

Question

我正在尝试使用映射将一组键映射到字符串映射到整数集。理想情况下，我想获取所有值集并将它们收集到一个集合中。

假设我有：

Map<String, List<Integer>> keyValueMap = new HashMap<>();
Set<String> keys = new HashSet<>();
Set<String> result = new HashSet<>();

我试过了：

result.addAll(keys.stream().map(key -> keyValueMap.get(key)).collect(Collectors.toSet());

这给我带来了一个错误，说 addAll() 不适用于 Set> 类型。我尝试用 flatMap() 替换 map()，但如果这是解决方案，我似乎无法获得正确的语法。

实现这项工作的正确语法是什么？

谢谢！

Answer 1

看起来result 的类型应该是Set<Integer> 而不是Set<String>。

---

使用您的 sn-p，您尝试在 Set<Integer> 上调用 Set#addAll，但传递的参数是 Set<List<Integer>>，它无法编译。

为了改善您的问题，一种解决方案是使用flatMap 而不是map：

result.addAll(keys.stream()
                  .flatMap(key -> keyValueMap.get(key).stream())
                  .collect(Collectors.toSet()));

一个逻辑上等价的sn-p是：

result.addAll(keys.stream()
                  .map(keyValueMap::get)
                  .flatMap(List::stream)
                  .collect(Collectors.toSet()));

---

另一个解决方案是使用Map#values:

result.addAll(keyValueMap.values().stream()
                         .flatMap(List::stream)
                         .collect(Collectors.toSet()));