scala reduce fold 化简 折叠

scala中集合类iterator特质的化简和折叠方法 
c.reduceLeft(op)这样的调用将op相继应用到元素,如: 
这里写图片描述

val a = List(1,7,2,9)
val a1 = a.reduceLeft(_ - _)//      ((1-7) - 2) - 9 = -17

 这里写图片描述

c.foldLeft(0)(_ * _)方法 
这里写图片描述

    val a2 = a.foldLeft(0)(_ - _) //    0-1-7-2-9 = -19

对于foldLeft方法还有一种简写,这种写法的本意是让你通过/:来联想一棵树的样子 
对/:操作符来说,初始值是第一个操作元本题中是0,:后是第二个操作元a

    val a3 = (0 /: a)(_ - _) //      等价于a.foldLeft(0)(_ - _)

scala同样也提供了foldRight或:\的变体,计算 
这里写图片描述

 val a4 = a.foldRight(0)(_ - _)//    1-(7-(2-(9-0))) = -13
 val a5 = (a :\ 0)(_ - _)   //    等价于a.foldRight(0)(_ - _)

好像上面的方法都没什么用,但折叠fold有时候可以代替循环,例如: 
统计每个字母中字符串出现的次数

object aiguigu {
  def main(args: Array[String]): Unit = {
    val tmp = "Mississippi"
    val frep = collection.mutable.Map[Char, Int]()
    for (c <- tmp)
      frep(c) = frep.getOrElse(c, 0) + 1

    val result = (Map[Char, Int]() /: tmp) // 使用 /: foldLeft()()来统计次数
    {
      (m, c) => m + (c -> (m.getOrElse(c, 0) + 1))
    }

    val result1 = tmp.foldLeft(Map[Char, Int]()) // 使用/: foldLeft()()来统计次数
    {
      (m, c) => m + (c -> (m.getOrElse(c, 0) + 1))
    }
    //      任何while循环都可用fold来代替
    println(frep, result, result1) //(Map(M -> 1, s -> 4, p -> 2, i -> 4),Map(M -> 1, i -> 4, s -> 4, p -> 2))
  }
}

这是使用fold的步骤:在每一步,将频率映射和新遇到的字母结合在一起,产生一个新的频率映射。这就是折叠: 

è¿éåå¾çæè¿°
 
fold的初始值和操作符是分开定义的柯里化参数,这样scala就可以根据类型来推断操作符的类型定义。 
eg:该函数的初始值是String,因此操作符的类型应该是(String,Int) => String的函数

    val a = List(1,7,2,9)
    val a7 = a.foldLeft("")(_ + _)
    val a8 = a.foldRight("")(_+_)
    val a9 = a.fold(0)(_+_)
    println(a7,a8,a9) //    (1729,1729,19)

 

 

 

 

 

 

posted @ 2019-11-01 18:08  sowhat1412  阅读(160)  评论(0编辑  收藏  举报