时间戳

首页 > 解决方案 > 使用尾递归和匹配表达式对 Scala 中的选项列表求和

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

问题描述

我必须使用匹配表达式制作一个尾递归程序，它将打印List[Option[Double]] fe类型的列表的总和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)

我只知道如何仅对 Some(value) 列表求和，不知道如何仅对正论点求和。

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))))

标签： scala

解决方案

您可以为此使用模式匹配保护：

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)

推荐阅读