数值参数化类型的初始值 (0)



考虑一下,我想实现一些函数,它将Long => T应用于整数范围a..b并累积类型T的结果(这是练习,而不是寻找有效的解决方案)

def sum[T <: Number](f: Long => T)(a: Long, b: Long): T = {
  def loop(acc: T, n: Long): T = 
    if (n > b)
      acc
    else
      loop(acc + f(n), n + 1)
  loop(0, a)
}

它在loop(0,抱怨方面存在缺陷

error: type mismatch;
 found   : Int(0)
 required: T
     loop(0, a)

我明白为什么,但是在这里给出数字T类型0的选项是什么?当然,如果有的话。

您应该将Numeric类型类用于泛型T。 这将使您能够访问方法zeroplus(因为每个Numeric都必须定义这些方法),这些方法将允许您通常执行求和。

def sum[T](f: Long => T)(a: Long, b: Long)(implicit num: Numeric[T]): T = {
  def loop(acc: T, n: Long): T =
    if (n > b)
      acc
    else
      loop(num.plus(acc, f(n)), n + 1)
  loop(num.zero, a)
}

顺便说一句:这就是 Scala 内置的 sum 方法所做的:

// from TraversableOnce
def sum[B >: A](implicit num: Numeric[B]): B = foldLeft(num.zero)(num.plus)

最新更新