scala练习——fold函数

题目

用fold方法将数组中的每个列表相同下标的元素进行聚合。

数据：Array(List(2,3,4), List(1,2,3), List(4,5,6))

结果为：Array(7,10,13)

解析

fold函数，英文翻译意为折叠，其源代码如下：

def fold[A1 >: A](z: A1)(op: (A1, A1) => A1): A1 = foldLeft(z)(op)
//该函数有两个参数，且后一个参数为一个函数，两个参数是采用柯里化联合到一起的。后面调回的是
//foldLeft()函数，其源码如下

def foldLeft[B](z: B)(op: (B, A) => B): B = {
    var result = z
    this foreach (x => result = op(result, x))
    result
  }

为了解其过程不妨测试如下代码：

val array: Array[List[Int]] = Array(List(2,3,4), List(1,2,3), List(4,5,6))
array.fold(0)((a1, a2) => {
     println(s"a1 : $a1   ---  a2 : $a2")
})

运行结果

这里第一次的a1为零，是第一个参数，a2为第一个list；第二次的a1是空元组，a2是第二个list；第三次的a1是空元组，s2是第三个list。据此可以推出fold是在遍历，并将之聚合为一个返回值。

因此，可以这样解题

val array: Array[List[Int]] = Array(List(2,3,4), List(1,2,3), List(4,5,6))
array.fold(List(0,0,0))((a1, a2) => {
    //在这里我曾尝试直接用a1加a2并返回，但是a1和a2是List，是不可变的序列
     val buffer: mutable.Buffer[Int] = a1.toBuffer
     buffer(0) = a1(0) + a2(0)
     buffer(1) = a1(1) + a2(1)
     buffer(2) = a1(2) + a2(2)
     buffer.toList
}).foreach(println)

结果