使用Java 8流API的累计和

Question

我有一个整数列表，比如list1，我想获得另一个列表list2，它将包含从开始到当前索引的累计总和。我如何使用流API Java8来实现这一点？

List<Integer> list1 = new ArrayList<>();
list1.addAll(Arrays.asList(1, 2, 3, 4));
List<Integer> list2 = new ArrayList<>();
// initialization
list2.add(list1.get(0));
for(int i=1;i<list1.size();i++) {
// increment step
    list2.add(list2.get(i-1) + list1.get(i));
}

如何将上面的命令式代码更改为声明式代码？

list2 should be [1, 3, 6, 10]

Answer 1

流不适合这种任务，因为它涉及到状态（累积部分和）。相反，您可以使用arrays.parallelprefix:

Integer[] arr = list1.toArray(Integer[]::new);

Arrays.parallelPrefix(arr, Integer::sum);

List<Integer> list2 = Arrays.asList(arr);

首先使用collection.toArray将list1复制到一个数组中，自JDK11以来，它就可用了。如果您还没有使用Java11，可以用传统的toArray调用替换第一行：

Integer[] arr = list1.toArray(new Integer[0]);

此解决方案不使用流，但它是声明性的，因为arrays.parallelprefix将累积操作作为参数接收（本例中为integer::sum)。

时间复杂度是O(N)，尽管可能存在一些与建立并行处理所需的基础结构相关的不小的恒定成本。然而，根据文件：

对于大型数组，并行前缀计算通常比顺序循环更有效

因此，似乎值得尝试一下这种方法。