algorithm - 命令式代码的惯用 Scala 解决方案

Question

在“惯用的”Scala 中表达这个函数有什么想法。或者更准确地说，有没有一种方法可以在不牺牲可读性的情况下删除本地变量？

def solve(threshold: Int)(f: Int => Int): Int = {
  var sum = 0
  var curr = 0
  while(sum < threshold) {
   sum += f(curr)
   curr += 1
  }
  curr
}

我唯一能想出的就是这个，但我认为它更长且可读性差。

def solve2(threshold: Int)(f: Int => Int): Int = {
  val resultIterator = Iterator.iterate (0, 0) { case (curr, sum) =>
    (curr + 1, sum + f(curr))
  }
  (resultIterator find (_._2 >= threshold)).get._1
}

Answer 1

def solve(threshold: Int)(f: Int => Int): Int = {
  Iterator.from(0).map(f).scanLeft(0)(_ + _).indexWhere(threshold <=)
}

在我看来，循环版本更清晰。