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

Question

我想修改这里给出的代码：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))}
}

也可以与 Long、Float 和 Double 一起使用（因为这些是 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))}
}

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

Answer 1

你想要一个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。

Answer 2

另一种方法是使用Numeric 的最小值/最大值：

def minMax[T](s: Seq[T])(implicit num: Numeric[T]): (T, T) = {
    if (s.isEmpty) throw new java.lang.UnsupportedOperationException("seq is empty")
    s.foldLeft((s.head, s.head)) {case ((min, max), e) => (num.min(min, e), num.max(max, e))}
  }