Scala foldLeft 而某些条件为真

Question

如何在 Scala 中模拟以下行为？即在满足蓄能器的某些特定条件时继续折叠。

def foldLeftWhile[B](z: B, p: B => Boolean)(op: (B, A) => B): B

例如

scala> val seq = Seq(1, 2, 3, 4)
seq: Seq[Int] = List(1, 2, 3, 4)
scala> seq.foldLeftWhile(0, _ < 3) { (acc, e) => acc + e }
res0: Int = 1
scala> seq.foldLeftWhile(0, _ < 7) { (acc, e) => acc + e }
res1: Int = 6

更新：

根据@Dima 的回答，我意识到我的意图有点副作用。所以我让它与 同步takeWhile，即如果谓词不匹配，则不会有任何进展。并添加更多示例以使其更清晰。（注意：这不适用于Iterators）

Answer 1

首先，请注意您的示例似乎是错误的。如果我正确理解你的描述，结果应该是1（_ < 3满足谓词的最后一个值），而不是6

最简单的方法是使用return声明，这在 scala 中非常不受欢迎，但我想，为了完整起见，我会提到它。

def foldLeftWhile[A, B](seq: Seq[A], z: B, p: B => Boolean)(op: (B, A) => B): B = foldLeft(z) { case (b, a) => 
   val result = op(b, a) 
   if(!p(result)) return b
   result
}

由于我们想避免使用 return，scanLeft可能是一种可能性：

seq.toStream.scanLeft(z)(op).takeWhile(p).last

这有点浪费，因为它累积了所有（匹配）结果。您可以使用iterator而不是toStream来避免这种情况，但由于某种原因Iterator没有这样.last做，因此，您必须明确地扫描它额外的时间：

 seq.iterator.scanLeft(z)(op).takeWhile(p).foldLeft(z) { case (_, b) => b }