java - Scala中的排序方法

Question

我想为 scala Array 扩展一个排序方法，它可以返回原始索引。

下面是我写的代码:

object ArrayExtend {
implicit class ArrayExtension(val array: Array[Int]) {
    def argSort(): Array[Int] = {
                var tupleArray = for (i <- 0 until array.length) yield (i, array(i))     

                    tupleArray = tupleArray.sortWith((x,y)=> x._2 < y._2)

                    val result = for((index,_) <- tupleArray) yield index

                    return result.toArray

    }
}


def main(args:Array[String]){
val array = Array(5,4,3,2,0)
for(i <- array.argSort()) print(i)
}

此代码适用于 Array[Int]，我如何将此方法扩展到所有不同类型的数组？我在 AnyRef 类中没有找到任何比较方法。

Answer 1

如果你需要那个类型就够了 T可以隐式转换为 Ordered[T] .通常它会通过使用 (implicit converter: T => Ordered[T]) 向类或方法参数添加隐式转换器来完成。 ，但是 scala 提供了 View 边界 <% 正是针对这种情况。摘自本书:您可以将“T <% Ordered[T]”想象成一句话。 “我可以使用任何 T，只要 T 可以被视为 Ordered[T]。”

所以代码看起来像这样:

implicit class ArrayExtension[T <% Ordered[T]](val array: Array[T]) {
  def argSort = array.zipWithIndex.sortBy(_._1).map(_._2).toArray
}

val array = Array(5,4,3,2,0)
val array2 = Array(5f,4f,3f,2f,0f)

array.argSort foreach print //43210
println
array2.argSort foreach print //43210