如何使用java流对作为HashMap值的ArrayList的元素求和？

Question

我有一个带有整数和一些元素对象的 ArrayList 的 HashMap。 元素对象用价格和数量来描述。 我想遍历每个 ArrayList 中的所有这些元素，通过调用每个元素element.price()总结它们，并创建一个新的 HashMap，其中包含代表每个数组列表总和的旧键和新值。 新哈希映射的键应保持不变。 尝试使用流来做到这一点。

public static HashMap<Integer, Double> findIncomes(HashMap<Integer, ArrayList<Element>> mapa){


    Map<String, Double> m = mapa.entrySet().stream().flatMap()

    return m;
}

Answer 1

我想到的第一个解决方案是使用mapToDouble和sum 。

那看起来像这样：

public static HashMap<Integer, Double> findIncomes(HashMap<Integer, List<Element>> mapa) {
    HashMap<Integer, Double> sumsByKey = new HashMap<>();
    mapa.entrySet().stream().forEach(entry -> sumsByKey.put(entry.getKey(), entry.getValue().stream().mapToDouble(element -> element.getPrice()).sum()));
    return sumsByKey;
}

但是当总结 1.5d、5.4d 和 6.7d 时，结果是 13.600000000000001。

因此我必须记住：使用双打进行计算时，通常最好使用BigDecimal 。

因此，更准确的解决方案可能如下所示：

public static HashMap<Integer, Double> findIncomes(HashMap<Integer, ArrayList<Element>> mapa){
    HashMap<Integer, Double> sumsByKey = new HashMap<>();
    mapa.entrySet().stream().forEach(entry -> sumsByKey.put(entry.getKey(),
        entry.getValue().stream().map(element -> BigDecimal.valueOf(element.getPrice())).reduce(BigDecimal.ZERO, BigDecimal::add).doubleValue()));
    return sumsByKey;
}

由于流中的流并不是真正可读的，因此进一步重构它可能是有意义的。

Answer 2

你总是需要一张新地图。 您不能使用不同类型更改同一地图。 像这样的东西可以完成这项工作，

public static HashMap<Integer, Double> findIncomes(HashMap<Integer, ArrayList<Element>> mapa) {

        final HashMap<Integer, Double> m = new HashMap<>();
        mapa.entrySet().stream().forEach(entry -> m.put(entry.getKey(), Double.valueOf(entry.getValue().stream().mapToDouble(Element::price).sum())));
        return m;
    }