Ruby：如何有效地对两个哈希数组执行“左连接”

Question

我正在尝试左加入以下哈希数组：

input:
a = [{id: 1, name: 'Bob'}, {id: 2, name: 'Jack'}, {id: 3, name: 'Tom'}]
b = [{id: 3, age: 12}, {id: 2, age: 7}]
output:
[{id: 1, name: 'Bob', age: nil}, {id: 2, name: 'Jack', age: 7}, {id: 3, name: 'Tom', age: 12}] 

目前我正在做一些事情：

a.map do |x|
  {
   id:   x[:id], 
   name: x[:name], 
   age:  (b.detect{|y| x[:id] == y[:id]} || {age: nil}).fetch(:age)
  }
end

可以，但是数据量大的时候超级慢。

有没有更好的方法来更高效地执行“加入”操作？

Answer 1

[a, b].map { |a| a.group_by { |e| e[:id] } }
      .reduce do |a, b|
          a.merge(b) { |_, v1, v2| v1.first.merge v2.first }
        end.values
      .map do |e|
         Array === e ? {age:nil, name:nil}.merge(e.first) : e
      end

整个准备步骤需要O(N)，然后合并以O(N) 完成，最后完成需要O(N)。

Answer 2

h = b.each_with_object({}) { |g,h| h[g[:id]] = g[:age] }
  #=> {3=>12, 2=>7} 
a.map { |g| g.merge(age: h[g[:id]]) }
  #=> [{:id=>1, :name=>"Bob",  :age=>nil},
  #    {:id=>2, :name=>"Jack", :age=>7},
  #    {:id=>3, :name=>"Tom",  :age=>12}] 

如果a要原地修改，则将第二行改为

a.each { |g| g[:age] = h[g[:id]] }

a #=> [{:id=>1, :name=>"Bob",  :age=>nil},
  #    {:id=>2, :name=>"Jack", :age=>7},
  #    {:id=>3, :name=>"Tom",  :age=>12}]