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),你可以通过累加器和列表(包含过去的累加器)折叠列表来自己实现它.结果你采取后者.