Scala:什么是在泛型类中进行数值运算的最佳方法?

时间:2009-12-08 23:58:00

标签: generics scala numerical

在Scala中,我希望能够编写使用>,/,*等运算符的泛型类,但是我没有看到如何约束T这样就可以了。

我研究了使用Ordered [T]约束T,但这似乎不起作用,因为只有RichXXX(例如RichInt)扩展它,而不是Int等。我也看过Numeric [T],这只在Scala中可用2.8

这是一个具体的例子:

class MaxOfList[T](list: List[T] ) {
  def max = {
    val seed: Option[T] = None

    list
      .map( t => Some(t))
      // Get the max      
      .foldLeft(seed)((i,m) => getMax(i,m) )
  }

  private def getMax(x: Option[T], y: Option[T]) = {
    if ( x.isDefined && y.isDefined )
      if ( x > y ) x else y
    else if ( x.isDefined )
      x
    else
      y
  }
}

这个类不会编译,因为有许多Ts不支持>等。

思想?

现在我使用了MixIn特性来解决这个问题:

/** Defines a trait that can get the max of two generic values
 */
trait MaxFunction[T] {
  def getMax(x:T, y:T): T
}

/** An implementation of MaxFunction for Int
 */
trait IntMaxFunction extends MaxFunction[Int] {
  def getMax(x: Int, y: Int) = x.max(y)
} 

/** An implementation of MaxFunction for Double
 */
trait DoubleMaxFunction extends MaxFunction[Double] {
  def getMax(x: Double, y: Double) = x.max(y)
} 

如果我们改变原始类,可以在实例化时混合使用。

P.S。 Mitch,受到你重写getMax的启发,这是另一个:

  private def getMax(xOption: Option[T], yOption: Option[T]): Option[T] = (xOption,yOption) match {
    case (Some(x),Some(y)) => if ( x > y ) xOption else yOption
    case (Some(x), _) => xOption
    case _ => yOption
  }

3 个答案:

答案 0 :(得分:9)

您可以使用View Bounds

简而言之,def foo[T <% U](t: T)是一个函数,它将采用任何或者可以隐式转换为U的函数。由于Int可以转换为RichInt(包含您想要的方法),这是一个很好的使用示例。

class MaxOfList[T <% Ordered[T]](list: List[T] ) {
  def max = {
    val seed: Option[T] = None
    list.foldLeft(seed)(getMax(_,_))
  }

  private def getMax(xOption: Option[T], y: T) = (xOption, y) match {
    case (Some(x), y) if ( x > y ) => xOption
    case (_, y) => Some(y)
  }
}

PS - 我重写了你的getMax(...)方法来比较值而不是选项本身,并使用模式匹配而不是isDefined(...)

PPS - Scala 2.8将具有可能有用的数字特征。 http://article.gmane.org/gmane.comp.lang.scala/16608


<强>附录

只是为了咯咯笑,这是超级紧凑版本,它完全消除了getMax方法:

class MaxOfList[T <% Ordered[T]](list: List[T] ) {
  def max = list.foldLeft(None: Option[T]) {
      case (Some(x), y) if ( x > y ) => Some(x)
      case (_, y) => Some(y)
  }
}

又一个附录

此版本对大型列表更有效...避免为每个元素创建一些(x):

class MaxOfList[T <% Ordered[T]](list: List[T] ) {
  def max = {
    if (list.isEmpty) None
    else Some(list.reduceLeft((a,b) => if (a > b) a else b))
  }
}

最后一个,我保证!

此时,您可以抛弃该类并使用函数:

  def max[T <% Ordered[T]](i: Iterable[T]) = {
    if (i.isEmpty) None
    else Some(i.reduceLeft((a,b) => if (a > b) a else b))
  }

答案 1 :(得分:2)

scala 2.8中的Numeric[T]

scala> case class MaxOfList[T : Numeric](list: List[T]) {
     |     def max = if(list.isEmpty) None else Some(list.max)
     | }
defined class MaxOfList

scala> MaxOfList(1::2::3::9::7::Nil).max

res1: Option[Int] = Some(9)

scala> MaxOfList(1.5::3.9::9.2::4.5::Nil).max

res2: Option[Double] = Some(9.2)

scala> MaxOfList(Nil: List[Byte]).max

res3: Option[Byte] = None

scala>

答案 2 :(得分:1)

正如Mitch Blevins所回答的那样,视图边界是为什么做的。现在,虽然Numeric仅在Scala 2.8上添加,但并不依赖于任何Scala 2.8功能。人们可以在Scala 2.7上做同样的事情。

您可能对我为Rosetta Code撰写的一些通用代码段感兴趣: