scala способ определения функций, принимающих список различных числовых типов - PullRequest
Новая
       26

scala способ определения функций, принимающих список различных числовых типов

6 голосов
/ 10 февраля 2010

У меня следующая проблема: у меня есть функция, которая принимает 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].

Также обратите внимание на формулировку суммы в не-стиле scala в функции f, чтобы подтвердить необходимость использования 0 (или других констант) внутри самой функции (если я суммирую значения Int, я должен инициализировать s в 0, а не 0.0 .

Каков наилучший способ (без кода) заставить функцию работать как на Double, так и на Int?

(я видел кое-что о числовой характеристике 2.8, но я не очень уверен, как ее использовать ...)

Спасибо всем за помощь.

Ответы [ 2 ]

7 голосов
/ 10 февраля 2010

При использовании 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))

Теперь еще один пример, делающий сумму более масштабно: 

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))
4 голосов
/ 10 февраля 2010

В этом ответе используется числовая черта. 

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