arrays - Scala 与列表中当前元素和所有过去元素的部分求和

Question

我们有一个整数列表，例如:[1,4,5,6,6,7,9]。

这个想法是生成一个具有相同长度的列表，并求和直到当前元素，例如:[1,5,10,16,22,29,38]。

在 Java 世界中，它看起来像:

int sum = 0;
int[] table = {1,4,5,6,6,7,9}
int[] res = new int[table.length]
for(int i=0; i<table.length; i++) {
  sum += table[i]
  res[i] = sum
}

我知道存在更优雅、更高效的解决方案。我的问题是如何在 Scala 中以更实用的方式做这样的事情？

谢谢!

Answer 1

您正在寻找扫描组合器。

List(1,4,5,6,6,7,9).scanLeft(0)(_ + _)
res1: List[Int] = List(0, 1, 5, 10, 16, 22, 29, 38)

如果需要，请删除带有尾部的前导元素我不知道不采用初始值的扫描版本。对于这个人来说，复杂度是 O(n)，你可以通过一个累加器和一个列表(包含过去的累加器)折叠列表来自己实现它。您将后者视为结果。