ordenar una matriz de acuerdo con los elementos de otra matriz

Question

tengo una tabla de id

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

y tengo otro conjunto de objetos con ids en orden aleatorio

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

Ahora necesito para ordenar a2 según el orden de los identificadores en a1. Así a2 ahora debe convertirse en:

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

a1 podría ser [3, 2, 5, 4, 1] o en cualquier orden pero a2 debe corresponder al orden de ids en a1.

hago así:

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 

Pero esto todavía podría funcionar en un tiempo O (n^2) si el orden de los elementos de A2 es exactamente inversa de a1. ¿Puede alguien decirme la forma más eficiente de ordenar 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 

creo que el tiempo de ejecución será O (n)

Answer 2

Me sorprendería si algo es mucho más rápido que la manera obvia:

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

Answer 3

me gusta la respuesta aceptada , pero en ActiveSupport hay index_by que hace que crear el hash inicial sea aún más fácil. Ver Cleanest way to create a Hash from an Array

De hecho, usted puede hacer esto en una sola línea desde Enumerable apoya index_by así:

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

Answer 4

Inspirado por Eric Woodruff's Answer, me ocurrió con la siguiente solución de vainilla Ruby:

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

Documentación de método:

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