scala方法定义函数接受不同数字类型的List

时间:2010-02-10 08:26:36

标签: scala scala-2.8 function-parameter

我有以下问题:我有一个函数,它以List [Double]作为参数,对列表的元素执行一些算术运算,然后返回结果。我希望该函数也接受List [Int]。这是一个例子:

def f(l: List[Double]) = {
    var s = 0.0 
    for (i <- l)
        s += i
    s
}

val l1 = List(1.0, 2.0, 3.0)
val l2 = List(1, 2, 3)

println(f(l1))
println(f(l2))

当然第二个println失败,因为f需要List [Double]而不是List [Int]。

还要注意f函数中和的非scala样式公式,以证明需要在函数本身内使用0(或其他常量)(如果我求和Int值我必须初始化为0而不是0.0

在Double和Int上使用该函数的最佳方法(更少的代码)?

(我已经看到一些关于2.8数字特性的东西,我不太确定如何使用它......)

感谢大家的帮助。

2 个答案:

答案 0 :(得分:7)

使用scala 2.8并使用Numeric combine进行隐式转换,您的示例可以写为:

import Numeric._
def f[T](l: List[T])(implicit n: Numeric[T]):T = {
    var s = n.zero
    for (i <- l)
        s = n.plus(s, i)
    s
}

val l1 = List(1.0, 2.0, 3.0)
val l2 = List(1, 2, 3)

println(f(l1))
println(f(l2))

//or
def f2[T](l: List[T])(implicit n: Numeric[T]):T = {
 import n._
 var s = zero
 for (i <- l)
   s += i
 s
}
println(f2(l1))
println(f2(l2))

现在是另一个以更加scala方式进行求和的示例:

def sum[T](l:List[T])(implicit n: Numeric[T]):T = {
 import n._
 l.foldLeft(zero)(_ + _)
}

println(sum(l1))
println(sum(l2))

//or since 2.8 Seq include already a sum function
def sum[T](l:List[T])(implicit n: Numeric[T]):T = l.sum

println(sum(l1))
println(sum(l2))

答案 1 :(得分:4)

This answer使用数字特征。

import Numeric._
def f[A](l: List[A])(implicit numeric: Numeric[A]) = 
  l reduceLeft ((l,r) => numeric.plus(l, r))

或使用上下文界限:

def f[A : Numeric](l: List[A]) =
   l.reduceLeft((l,r) => implicitly[Numeric[A]].plus(l, r))