配列の結合について - 重複したもののみ結合する

Question

今、2つの配列の要素で重複したものだけ結合して、1つの配列にしたい処理を書いております。

arr1 = [1,2,3,4,5,6,7,8,9]
arr2 = [2,2,5,7,7,10,12]

得たい結果は重複した場合のみarr1に追加したいです。
[1,2,3,4,5,6,7,8,9,2,2,5,7,7]

arr2.each do |n|
  if  arr1.include?(n)
      arr1.push(n)
    end
end

上記のようなコードを自分なりに書いてみたのですが、結果が下記のようになり、思い通りの結果になりませんでした。

[1,2,3,4,5,6,7,8,9,2]
[1,2,3,4,5,6,7,8,9,2]
[1,2,3,4,5,6,7,8,9,5]
[1,2,3,4,5,6,7,8,9,7]
[1,2,3,4,5,6,7,8,9,7]

アドバイスをいただけますと幸いです。

Answer 1

arr1、 arr2 両方に含まれる要素は arr1 & arr2 で得る事ができます。

arr1 & arr2 # => [2, 5, 7]

と、いう事で Hello さんの書き方をベースにこれ(積集合)を使って書くと

arr2.each do |n|
  if (arr1 & arr2).include?(n)
    arr1.push(n)
  end
end
arr1 
# => [1, 2, 3, 4, 5, 6, 7, 8, 9, 2, 2, 5, 7, 7]

ですね。

1行で書くなら

arr1.concat(arr2.select{ |n| (arr1 & arr2).include?(n) })
# => [1, 2, 3, 4, 5, 6, 7, 8, 9, 2, 2, 5, 7, 7]

とも書けますね。

ちなみにどちらも arr2 の要素数だけ arr1 & arr2 をしてしまうので最初に別途変数に入れておいた方がいいですね。

---

もっとうまい書き方もあると思うので

• Ruby 2.3.0 リファレンスマニュアル > ライブラリ一覧 > 組み込みライブラリ > Arrayクラス
• Ruby 2.3.0 リファレンスマニュアル > ライブラリ一覧 > 組み込みライブラリ > Enumerableモジュール

この辺とにらめっこしてみてください。

Comments

[Hello]

ありがとうございます！
思い通りの結果が取得できました。

とっても勉強になりました。