scala练习——fold函数

最新推荐文章于 2021-12-02 14:56:32 发布

yeah_you_are

最新推荐文章于 2021-12-02 14:56:32 发布

阅读量708

点赞数 1

分类专栏： scala 文章标签： scala

本文链接：https://blog.csdn.net/yeah_you_are/article/details/117675419

版权

scala 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何利用fold方法将数组中的每个列表相同下标的元素进行聚合。通过示例代码解释了foldLeft函数的工作原理，并展示了如何在实际操作中处理不可变序列，最终实现将三个列表对应位置的元素相加，得到结果Array(7,10,13)。

摘要由CSDN通过智能技术生成

题目

用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)