【Ruby】二重のeach文

Question

超初歩的な質問で申し訳無いのですが、お答え頂ければと思います。
as = [1, 2, 3, 4, 5]
bs = [1, 4, 5]
という２つの配列があるときに、
cs = [1, nil, 3, nil, 5]
という配列を作るロジックを考えているのですが、以下のロジックを、よりスマートに書く方法は無いでしょうか？

cs = []
as.each_with_index do |a, i|
  bs.each do |b|
    if a == b
      cs << b
    end
  end
  if cs[i] == nil
    cs[i] = nil
  end
end

よろしくお願い致します。

Answer 1

期待する出力はもしかして、[1, nil, nil, 4, 5] では？
もしそうならこんな感じでどうでしょう？

as = [1, 2, 3, 4, 5] 
bs = [1, 4, 5]
cs = []
as.each do |a|
    cs << (bs.include?(a)? a: nil)
end

print cs

Comments

[Miracle]

正に実現したい実装でした。
includesメソッド一つで、こんなにも簡単に記述できるのですね！
勉強になりました！ありがとうございます！