rubyで配列を任意の数で分割する方法を教えてください

Question

ruby(railsでも可)で下記のような配列を任意の要素数で分割する方法を教えてください。

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

の配列を、任意の要素数が2,5,3だった場合に下記のように分けたいです。

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

今思いついてるのは

arr = [1,2,3,4,5,6,7,8,9,10]
offsets = [2, 5, 3]
offset = 0

offsets.each_with_object([]) do |n, result|
  result.push arr[offset..(offset + n - 1)]
  offset += n
end

という感じです。offsetsは数値を直書きしてますが実際は、任意の要素数が不定、かつ分割数も不定の変数になります。(任意の要素数は元の配列の長さを超えないこととします)
この場合、どのように記述すれば↑のコードより完結かつ高速に処理できるか教えていただきたいです。

何卒よろしくお願いします！

Answer 1

shiftで順番に取り出します。

arr = [1,2,3,4,5,6,7,8,9,10]
offsets = [2, 5, 3]
p offsets.each_with_object(arr.dup).map{|of,ar| ar.shift(of)}

で、arr(のコピー)に対して、arr.shift(2) arr.shift(5) arr.shift(3)を順に実行して、それを配列にして返します。

Answer 2

def bunkatsu(arr, cs)
  arr.each_with_index
    .chunk { |e, i| cs.find_index { |c| (i -= c) < 0 } }
    .map(&:last)
    .map { |a| a.map(&:last) }
end

arr = [1,2,3,4,5,6,7,8,9,10]
counts = [2, 5, 3]
p bunkatsu(arr, counts).to_a

こんな感じだと思います。

配列の分割にはEnumerable#chunkを使います。chunkはブロックでの値毎に配列を分割するというものです。では、chunkのブロックで何を計算するかというと、分割する要素数の書かれた配列において、何番目の所に来るかです。そこで、each_with_indexであらかじめインデックスをつけておいて、それを順番に要素数で引いていきます。もし、それが負の値なら、そこで終わりと言うことですが、負で無ければ次の所を見に行きます。(なんか説明がよくわからない感じになっていますが、cs.find_index { |c| (i -= c) < 0 } }がどのような値になるのかを手で計算してみると良いです。)あとは、いらない部分を削っていけば完成となります。

速度はO(m*n)なので、速いとは言えないです。インデックスを付けるのでは無く、カウント方式にすればO(n)にはなると思いますが、もう一つの要件である簡潔なコードを私には書けそうにはないため、他の人に任せます。

Answer 3

次のコードを研究してみてください。
div.ry

arr = [1,2,3,4,5,6,7,8,9,10]
offsets = [2, 5, 3]
# offsets = [2, 5, 3, 10]                                                                                
# offsets = [2, 5]                                                                                       

starts = offsets.each_with_object([0]){|x, mem| mem << mem[-1] + x}
puts("starts = #{starts.inspect}")

starts.each_cons(2){|v| p v}

ans = starts.each_cons(2).map{|x, y| arr[x..y - 1]}
print(ans)
puts

# 上を一行にまとめる                                                                                     
ans = offsets
        .each_with_object([0]){|x, mem| mem << mem[-1] + x}
        .each_cons(2)
        .map{|x, y| arr[x..y - 1]}
print(ans)
puts

実行例

offsets(各要素の長さ) を, 各要素の開始位置の配列に変換して、それをつかって結果の要素をつくりだしています。