Question

我有一个HashMaps矩阵，我试图根据给定的输入添加那些HashMaps的内容，这些输入将匹配＆＃34;外部＆＃34; HashMap的密钥。拿下面的SSCCE（有1个错误说明我的进度），有4个HashMaps的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);
    }
}

Answer 1

我认为从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);

Answer 2

作为你所拥有的一个延续（所以你可以看到差异），这就是你如何完成它：

total = map.entrySet()
    .stream()
    .filter(e -> keys_to_add.contains(e.getKey()))
    .map(Map.Entry::getValue)
    .map(Map::values)
    .flatMap(Collection::stream)
    .reduce(BigDecimal.ZERO, BigDecimal::add);

我同意Misha @，从keys_to_add

开始更有意义

使用流来添加多个哈希映射的内容

2 个答案: