ruby - 根据给定的顺序对数字数组进行排序

Question

我有两个数组。第一个数组包含排序顺序。第二个数组包含任意数量的元素。

我有一个属性，即第二个数组中的所有元素（按值）都保证在第一个数组中，并且我只使用数字。

A = [1,3,4,4,4,5,2,1,1,1,3,3]
Order = [3,1,2,4,5]

当我排序时A，我希望元素按以下指定的顺序出现Order：

[3, 3, 3, 1, 1, 1, 1, 2, 4, 4, 4, 5]

请注意，重复是公平的游戏。A 中的元素不应更改，只能重新排序。我怎样才能做到这一点？

Answer 1

>> source = [1,3,4,4,4,5,2,1,1,1,3,3]
=> [1, 3, 4, 4, 4, 5, 2, 1, 1, 1, 3, 3]
>> target = [3,1,2,4,5]
=> [3, 1, 2, 4, 5]
>> source.sort_by { |i| target.index(i) }
=> [3, 3, 3, 1, 1, 1, 1, 2, 4, 4, 4, 5]

Answer 2

如果（且仅当！）@Gareth 的回答太慢，请改为：

# Pre-create a hash mapping value to index once only…
index = Hash[ Order.map.with_index.to_a ] #=> {3=>0,1=>1,2=>2,4=>3,5=>4}

# …and then sort using this constant-lookup-time
sorted = A.sort_by{ |o| index[o] } 

---

基准测试：

require 'benchmark'

order = (1..50).to_a.shuffle
items = 1000.times.map{ order.sample }
index = Hash[ order.map.with_index.to_a ]

Benchmark.bmbm do |x|
  N = 10_000
  x.report("Array#index"){ N.times{
    items.sort_by{ |n| order.index(n) }
  }}
  x.report("Premade Hash"){ N.times{
    items.sort_by{ |n| index[n] }
  }}
  x.report("Hash on Demand"){ N.times{
    index = Hash[ order.map.with_index.to_a ]
    items.sort_by{ |n| index[n] }
  }}
end

#=>                      user     system      total        real
#=> Array#index     12.690000   0.010000  12.700000 ( 12.704664)
#=> Premade Hash     4.140000   0.000000   4.140000 (  4.141629)
#=> Hash on Demand   4.320000   0.000000   4.320000 (  4.323060)

Answer 3

另一种没有显式排序的可能解决方案：

source = [1,3,4,4,4,5,2,1,1,1,3,3]
target = [3,1,2,4,5]
source.group_by(&lambda{ |x| x }).values_at(*target).flatten(1)