scala - 使用尾递归和匹配表达式对 Scala 中的选项列表求和

Question

我必须使用匹配表达式创建一个尾递归程序，该程序将打印 List[Option[Double]] 类型的 List 的总和 f.e List(Some(1), ..., Some(n)) --> Some(1 + ... + n) 但它只对具有正数的列表求和，因此当 Some(d) 和 d < 0 时，它不算在内。

List(Some(1.0),Some(2.0), Some(-3.0) ---> Some(3.0)

当没有任何元素为正时，它应该返回None

sumuj(List(Some(2.0), Some(4.0), Some(-3.0), None, Some(-3.0), None, Some(1.0))) returns Some(7.0)

我只知道如何仅对某些(值)列表进行求和，不知道如何仅对正参数求和。

def sumuj(l: List[Option[Double]]): Option[Double] = {
  def helper(l: List[Option[Double]], acc: Double): Option[Double] = {
    l match {
      case head :: tail => head match {
        case Some(value)  => helper(tail, acc + value)
      }
      case Nil => Some(acc)
    }
  }
  helper(l, acc = 1)
}
println(sumuj(List(Some(2.0), Some(4.0), Some(-3.0))))

Answer 1

您可以使用 pattern match guard为此:

def sumPositives(l: List[Option[Double]]) = {
  @tailrec
  def sumPositives0(l: List[Option[Double]], acc: Double): Option[Double] =
    l match {
      case Some(head) :: tail if head > 0 => sumPositives0(tail, acc + head)
      case _ :: tail                      => sumPositives0(tail, acc)
      case Nil                            => Some(acc)
    }
  sumPositives0(l, acc = 1)
}

然后:

println(sumPositives(List(Some(2.0), Some(4.0), Some(-3.0))))

产量:

Some(7.0)