如何对 ArrayBuffer[Double] 进行排序并保存索引

Question

给定一个 ArrayBuffer[Double]，如何在维护其索引的同时对其元素进行排序，例如

val arr ArrayBuffer[Double] = ArrayBuffer(4,5.3,5,3,8,9)

结果必须是：

arrSorted = ArrayBuffer(3,4,5,5.3,8,9)
indices = Arraybuffer(3,0,2,1,4,5) //here the data structure doesn't matter, it can be Array, List, Vector, etc.

谢谢

Answer 1

这是单行的：

val (addSorted, indices) = arr.zipWithIndex.sorted.unzip

逐步进行，zipWithIndex 创建一个元组集合，索引作为每个元组中的第二个值：

scala> println(arr.zipWithIndex)
ArrayBuffer((4.0,0), (5.3,1), (5.0,2), (3.0,3), (8.0,4), (9.0,5))

sorted 按字典顺序对这些元组进行排序（这几乎肯定是你想要的，但你也可以使用 sortBy(_._1) 来明确你想按值排序的事实）：

scala> println(arr.zipWithIndex.sorted)
ArrayBuffer((3.0,3), (4.0,0), (5.0,2), (5.3,1), (8.0,4), (9.0,5))

unzip 然后将这个元组集合变成一个集合元组，你可以用 val (addSorted, indices) = ....

解构它

如何对 ArrayBuffer[Double] 进行排序并保存索引

How to sort an ArrayBuffer[Double] and save indices

scala