我有以下问题:我有一个函数,它接受 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 (或其他常量)(如果 i sum Int 值,我必须将 s 初始化为 0 而不是 0.0。
哪种方法是让函数在 Double 和 Int 上工作的最佳方法(更少的代码)?
(我已经看到一些关于 2.8 数字特征的东西,我不太确定如何使用它......)
谢谢大家的帮助。
使用 scala 2.8 并使用数字组合进行隐式转换,您的示例可以写为:
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))
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)