Trie un tableau selon les éléments d'un autre tableau

Question

Je dispose d'un tableau de ids

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

et j'ai un autre tableau d'objets avec ids dans un ordre aléatoire

a2 = [(obj_with_id_5), (obj_with_id_2), (obj_with_id_1), (obj_with_id_3), (obj_with_id_4)] 

Maintenant, je dois trier a2 selon l'ordre des ids en a1. devrait devenir maintenant si a2:

[(obj_with_id_1), (id_2), (id_3), (id_4), (id_5)] 

a1 peut-être [3, 2, 5, 4, 1] ou dans un ordre quelconque mais a2 doit correspondre à l'ordre de ids dans a1.

Je aime ceci:

a1.each_with_index do |id, idx| 
    found_idx = a1.find_index { |c| c.id == id } 
    replace_elem = a2[found_idx] 
    a2[found_idx] = a2[idx] 
    a2[idx] = replace_elem 
end 

Mais cela pourrait encore fonctionner dans un temps O (n^2) si l'ordre des éléments de a2 est inverse exactement de a1. Quelqu'un peut-il me dire s'il vous plaît la façon la plus efficace de trier A2?

Answer 1

hash_object = objects.each_with_object({}) do |obj, hash| 
    hash[obj.object_id] = obj 
end 

[1, 2, 3, 4, 5].map { |index| hash_object[index] } 
#=> array of objects in id's order 

Je crois que le temps d'exécution sera O (n)

Answer 2

Je serai surpris si quelque chose est beaucoup plus rapide que la manière évidente:

a2.sort_by{|x| a1.index x.id} 

Answer 3

J'aime la réponse acceptée , mais dans ActiveSupport, il y a index_by ce qui rend la création du hash initial encore plus facile. Voir Cleanest way to create a Hash from an Array

En fait, vous pouvez le faire en une seule ligne depuis Enumerable soutient index_by ainsi:

a2.index_by(&:id).values_at(*a1) 

Answer 4

Inspirée par Eric Woodruff's Answer, je suis venu avec la solution vanille Ruby suivante:

a2.group_by(&:object_id).values_at(*a1).flatten(1) 

Documentation de méthode:

• Enumerable#group_by
• Array#values_at
• Array#flatten