在 N 个哈希表的键之间执行 UNION 操作

Question

我有一个包含 N 个 hast 表的哈希表：

Map<Integer,Map<String,Double>

我需要创建一个包含内部映射的所有键的列表：

----------------
|     | a    2 |
| 100 | b    1 |    
|     | c    2 |
----------------
|     | a    2 |
| 101 | d    2 |     
---------------- 
|     | a    2 |
| 102 | b    1 |    
|     | e    2 |
----------------

列表 = {a,b,c,d,e}

这是我当前的代码：

Set<String> keys= new HashSet<>();
    map1.entrySet().forEach(e -> {
        keys.addAll(e.getValue().keySet());
    });

map1 包含数千个条目。

这是最佳方法吗？有人知道更快的方法吗？

Answer 1

您可以尝试使用以下代码：

Map<String, Double> innerMap = new HashMap<>();
innerMap.put("a", 2d);
innerMap.put("b", 2d);
innerMap.put("c", 2d);

Map<String, Double> innerMap2 = new HashMap<>();
innerMap2.put("a", 2d);
innerMap2.put("d", 2d);
innerMap2.put("e", 2d);

Map<Integer, Map<String, Double>> map = new HashMap<>();
map.put(100, innerMap);
map.put(101, innerMap2);

Set<String> collect = map.values()
                         .stream()
                         .parallel()
                         .map(Map::keySet)
                         .flatMap(Collection::stream)
                         .collect(Collectors.toSet());

---

很遗憾，如果它对性能有重大影响，您将不得不自己尝试。

---

这意味着您使用的是 Java 8。但是当您使用forEach() 方法时，我只是假设。

---

编辑：
请注意D. Kovács关于parallel()方法使用的评论：Details

Answer 2

基本上，您的方法最适合一般情况。如果您对 Map 结构、密钥分布等有额外的限制，您当然可以对其进行定制。但是对于一般情况，我没有看到更好的方法，也许是其他语法（例如，一直使用流：

map.entrySet()
   .stream()
   .map(Map.Entry::getValue)
   .flatMap(e -> e.keySet().stream())
   .collect(Collectors.toSet())

)

Answer 3

我认为您可以通过使用 flatMap() 来充分利用并行性

如果您可以估计所需的大小，您可能还希望考虑 ConcurrentHashMap，即Collections.newSetFromMap(new ConcurrentHashMap())

例子：

    final int EST_SIZE = 6_000_000;
    // Map<Integer,Map<String,Double>> map1;
    Set<String> keys = map1.values().stream().map(Map::keySet)
            .flatMap(Set::stream).parallel().unordered()
            .collect(Collector.of(
                    () -> Collections.newSetFromMap(
                        new ConcurrentHashMap<>(EST_SIZE * 4 / 3 + 1)
                    ),
                    Set::add,
                    (set1,set2) -> { 
                        set1.addAll(set2);
                        return set1;
                    },
                    Collector.Characteristics.CONCURRENT,
                    Collector.Characteristics.UNORDERED
                    ));

注意：以上代码是针对高度专业化场景的尝试优化。在决定之前对其进行测试并检查性能。在一般情况下，您应该更喜欢这样的东西：

    Set<String> keys = map1.values().stream().map(Map::keySet)
            .flatMap(Set::stream).collect(Collectors.toSet());