ruby：複数の配列から順番に要素を取得し、合計の要素が3になったら処理を終了する。

Question

要素の数がわからない配列が5個(array5、array4、array3、array2、array1)あるときに、array5から順にsampleメソッドを実行し、要素を全部で3つ取得するまで実行する場合、どのような式になるのでしょうか？

自分で考えたところ下記のような流れになるのだろうと思ったのですが複雑すぎてこんがらがって数時間経ってしまいました。

---

配列(array5)の要素数(n5)を確認。
n5=3以上なら、array5.sample(3)で終了。

n5=2以下なら、配列(array4)の個数(n4)を確認。
n5+n4=3以上なら、array5.sample(3)&array4.sample(3-n5)で終了。
n5+n4=2以上なら・・・・続く

---

ほぼ丸投げな状態になってしまいますが、お力をお貸しいただけると幸いです。

Answer 1

[array5,array4,array3,array2,array1].inject([]){|ret,ary| 
  break ret if ret.size == 3; ret + ary.sample(3-ret.size)
}

Comments

[no1knows]

いつもありがとうございます！期待する動作になりました！
今回、otnさんの追記の回答が個人的にわかりやすいと感じたため、ベストアンサーにさせていただきます。

Answer 2

少し無駄な処理をしますが、それぞれランダムに並び替えて連結して、先頭の3つを取ればいいかと。

N=3
[array5,array4,array3,array2,array1].map(&:shuffle).inject(&:+).take(N).sample(N)

array5のshuffleはそもそも無駄。array4以降は使われないかもしれないけどshuffleするという無駄。
単に連結してtake(3)だと、array5で足りない個数をarray4の先頭から使ってしまうので、array4以降はshuffleが必要です。

#追記
データの配列ごとの順序を守るとすると、最後のsampleをやめればいいのか。

N=3
[array5,array4,array3,array2,array1].map(&:shuffle).inject(&:+).take(N)

この場合は、array5のshuffleも無駄ではなくなります。

Comments

[otn]

あ、勘違いしてました。array5とarray4の要素は混ざっちゃダメなんですね。

[winterboum]

inject(&:+).flatten[0,3] なら？

[no1knows]

いつもありがとうございます！追記いただいたもので期待する動作になりました！
if文をつなげてどうにかやるしかないかと思っていたらこんなにシンプルな形になって驚きました。

Answer 3

***ruby
[array5,array4,array3,array2,array1].map(&:shuffle).concat.flatten[0..2]

は如何でしょう。

これは
　　[[1, 2], [3, 4], [5, 6]].concat は [[1, 2], [3, 4], [5, 6]] となります。
　　[[1, 2], [3, 4], [5, 6]].concat.flatten は [1, 2, 3, 4, 5, 6] となります。
ということが基本の考え方になってます。

arrayX が nil になることがあるのなら、 compact をしてから [0..2]として 先頭の要素を取り出すようにすると良いかもしれません。