给定一个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.
由于
答案 0 :(得分:10)
这是一个单行:
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) = ...
解构。