如何将类型限制为 Scala 中的特定类型

时间:2021-05-21 14:06:18

标签: scala generics numeric

我想修改这里给出的代码:Find min and max elements of array

def minMax(a: Array[Int]) : (Int, Int) = {
  if (a.isEmpty) throw new java.lang.UnsupportedOperationException("array is empty")
  a.foldLeft((a(0), a(0)))
  { case ((min, max), e) => (math.min(min, e), math.max(max, e))}
}

还可以使用 LongFloatDouble(因为这些是 scala.math.min/max 接受的类型。我试过:

def getMinAndMax[@specialized(Int,Long,Float,Double) T](x: Seq[T]) : (T, T) = {
  if (x.isEmpty) throw new java.lang.UnsupportedOperationException("seq is empty")
  x.foldLeft((x.head, x.head))
  { case ((min, max), e) => (math.min(min, e), math.max(max, e))}
}

但这也不能编译。有什么建议吗?

2 个答案:

答案 0 :(得分:5)

您想要一个 typeclass。具体来说,在这种情况下,您需要来自 stdlib 的 Ordering

// It is more idiomatic to return an Option rather than throwing an exception,
// that way callers may decide how to handle that case.
def getMinAndMax[T : Ordering](data: IterableOnce[T]): Option[(T, T)] = {
  import Ordering.Implicits._ // Provides the comparison operators: < & >

  data.iterator.foldLeft(Option.empty[(T, T)]) {
    case (None, t) =>
      Some((t, t))
    
    case (current @ Some((min, max)), t) =>
      if (t < min) Some((t, max))
      else if (t > max) Some((min, t))
      else current
  }
}

您可以看到运行 here 的代码。

答案 1 :(得分:1)

另一种方法是使用 0ICPRP900 0.006641 0ICPRP 900 0.01 0ICPRP A&B_900 0.641 中的最小值/最大值:

Numeric