实现折叠与理解

Question

在斯卡拉如何将fold作为for-comprehension实现？我看到唯一的方法是使用一些递归调用？这是一个失败的尝试，不知道如何做到这一点？什么是执行fold为for-comprehension

val nums = List(1,2,3)       
    nums.fold(0)(_+_)        
    def recFold(acc: Int = 0): Int = { 
    (for { 
     a <- nums 
     b = recFold(a + acc) 
    } yield b).head 
    }            
    recFold(0) //Stack overflow 

Answer 1

如果你真的想使用for，你不需要递归，但你需要一个可变变量：

val nums = List(1,2,3) 

def recFold(zero: Int)(op: (Int, Int) => Int): Int = { 
    var result: Int = zero 
    for { a <- nums } result = op(result, a) 
    result 
} 

recFold(0)(_ + _) // 6 

这是非常相似，如何foldLeft在TraversableOnce实际执行：

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

Answer 2

折可以实现从右到左或从左到右两种方式的最佳途径。无需使用for加递归。递归就够了。

def foldRight[A, B](as: List[A], z: B)(f: (A, B) => B): B = { 
    as match { 
     case Nil => z 
     case x :: xs => f(x, foldRight(xs, z)(f)) 
    } 
    } 

    @annotation.tailrec 
    def foldLeft[A, B](as: List[A], z: B)(f: (A, B) => B): B = { 
    as match { 
     case Nil => z 
     case x :: xs => foldLeft(xs, f(x, z))(f) 
    } 
    }