Sắp xếp một dãy số dựa trên một thứ tự nhất định

Question

Tôi có hai mảng. Mảng đầu tiên chứa thứ tự sắp xếp. Mảng thứ hai chứa một số phần tử tùy ý.

Tôi có thuộc tính rằng tất cả các phần tử (giá trị khôn ngoan) từ mảng thứ hai được đảm bảo nằm trong mảng đầu tiên và tôi chỉ làm việc với các số.

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

Khi tôi loại A, tôi muốn các yếu tố xuất hiện theo thứ tự quy định của Order:

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

Lưu ý rằng các bản sao là trò chơi công bằng. Các yếu tố trong A không nên thay đổi, chỉ được sắp xếp lại. Tôi có thể làm cái này như thế nào?

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

Nếu (và chỉ khi!) @ Câu trả lời Gareth của hóa ra là quá chậm, thay vì đi với:

# 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] } 

---

Benchmarked:

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

Một giải pháp khả thi mà không phân loại rõ ràng:

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)