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

时间:2012-09-25 19:35:04

标签: scala generics

考虑一下,我想实现一些函数,它将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的选项是什么?如果有的话,当然。

1 个答案:

答案 0 :(得分:2)

您应该使用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)