java - 使用流来添加多个哈希图的内容

我有一个HashMaps矩阵，我试图根据给定的输入将这些HashMaps的内容相加，该输入将匹配“外部” HashMap的键。以下面的SSCCE(带有1个错误，它说明了我的进度)，HashMaps有4个HashMap行，我想在第1行和第3行中添加HashMaps的内容。

我当然可以用旧的Java 7方法做到这一点。但是我想使用Java 8流来执行此操作，而不必进行迭代。

public class HashMapCollector {
    public static void main(String[] args) {
        HashMap<Integer, HashMap<Integer, BigDecimal>> map = new HashMap<>();
        map.put(1, new LinkedHashMap<>()); // row 1
        map.put(2, new LinkedHashMap<>()); // row 2
        map.put(3, new LinkedHashMap<>()); // row 3
        map.put(4, new LinkedHashMap<>()); // row 4

        // row 1
        map.get(1).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(1).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(1).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(1).merge(4, BigDecimal.TEN, BigDecimal::add);

        // row 2
        map.get(2).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(2).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(2).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(2).merge(4, BigDecimal.TEN, BigDecimal::add);

        // row 3
        map.get(3).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(3).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(3).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(3).merge(4, BigDecimal.TEN, BigDecimal::add);

        // row 4
        map.get(4).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(4).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(4).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(4).merge(4, BigDecimal.TEN, BigDecimal::add);

        // Using a HashSet to contain the keys of the rows I want
        HashSet<Integer> keys_to_add = new HashSet<>();
        keys_to_add.add(1);
        keys_to_add.add(3);

        // Java 7 boring vanilla iteration
        BigDecimal total = BigDecimal.ZERO;
        for(Entry<Integer, HashMap<Integer, BigDecimal>> entry : map.entrySet()) {
            if(keys_to_add.contains(entry.getKey())) {
                for(BigDecimal bd : entry.getValue().values()) {
                    total = total.add(bd);
                }
            }
        }

        System.out.println(total);

        // Java 8ish method -- still some boring iteration
        total = BigDecimal.ZERO;
        for(Entry<Integer, HashMap<Integer, BigDecimal>> entry : map.entrySet()) {
            if(keys_to_add.contains(entry.getKey())) {
                total = total.add(entry.getValue().values().stream().reduce(BigDecimal.ZERO, BigDecimal::add));
            }
        }

        System.out.println(total);


        // Java 8 method -- no iteration
        total = BigDecimal.ZERO;
        map.entrySet().stream().filter(e -> keys_to_add.contains(e.getKey())).map(Map.Entry::getValue). // This is as far as I got

        System.out.println(total);
    }
}

最佳答案

我认为从keys_to_add开始而不是遍历所有条目并检查键是否在keys_to_add中更有意义。

BigDecimal total = keys_to_add.stream()
        .filter(map::containsKey)
        .map(map::get)
        .map(Map::values)
        .flatMap(Collection::stream)
        .reduce(BigDecimal::add)
        .orElse(BigDecimal.ZERO);

或者，如果您希望保留垂直空间，

BigDecimal total = keys_to_add.stream()
        .filter(map::containsKey)
        .flatMap(k -> map.get(k).values().stream())
        .reduce(BigDecimal.ZERO, BigDecimal::add);