2個ずつ追加

Question

rubyを使って2こずつ追加して比較したいと考えています。
現在のコードだと
[[[1, 2], [3, 4]], [[3, 4], [5, 6]], [[5, 6], [7, 8]],[[7, 8],[9, 10]]]
このような結果になっています。

そこを
[[[1, 2], [3, 4]], [[1, 2, 3, 4], [5, 6]], [[1, 2, 3, 4, 5, 6], [7, 8]],[[1, 2, 3, 4, 5, 6, 7, 8],[9, 10]]]
このようにするにはどのように直せばよいですか？
簡単なことかもしれませんがどうかお助けください。

def input(tw,split_num)
	
	ar=[]
	ar_tmp=[]
	ar_result=[]
	
	for i in 1..tw.count do
		if i%split_num != 0 then
			ar_tmp.push(tw[i-1])
		else
			ar_tmp.push(tw[i-1])
			ar.push(ar_tmp)
			ar_tmp=[]
		end
	end
	ar_tmp=[]
	
	for i in 0..ar.count-2 do
		ar_result.push([ar[i],ar[i+1]])
	end
	
	return ar_result
	
end

ar=[1,2,3,4,5,6,7,8,9,10]
p input(ar,2)

Answer 1

def input(tw, split_num)
  tw.each_slice(split_num).inject([[[], []]]) do |result, item|
    result << [result.last.flatten, item]
  end.drop(2)
end
ar = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
p input(ar, 2)

今のコードを直す方向は少し難しいので、全く作り替えました。以下、解説です。

1. N個ずつはEnumerable#each_sliceを使おう。

each_sliceはN個ずつまとめるという便利なメソッドです。Enumerableのメソッドなので、Arrayだけではなく、eachが使えるオブジェクトであれば何でも使えます。例えば、[1, 2, 3, 4, 5, 6].each_slice(2).to_aとすれば、[[1, 2], [3, 4], [5, 6]]になります。ブロックを付けなければEnumerableを返すので、そのまま次の計算に持って行くことも出来ます。

1. 要素を見ていく度に増えていく場合はEnumerable#injectを使おう。

injectはreduceという別名があります。他の言語をしている人にはreduceやfoldlといった場合の方が通じます。実際にどう増えていくかを見ると、何をしているかがわかりやすいでしょう。

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

上のはresultを毎回出力したものです。それぞれが、前の配列をフラット(flatten)したものと今回の要素のペアとした配列になっています。なので、まいかいそれをresultに追加していけば良いとなります。初期値は最初の要素でエラーとならないように空のデータを入れています。二番目の要素も結果には出さないので、最後にdrop(2)で最初の二つを削除して、目的の配列を得ています。

Answer 2

問題を 2 個づつでなく、n 個ずつ(n は 1 以上) と、一般化してみました。
gen.rb

def generate(ary, n)
  ans = []
  (1...(ary.length / n)).each do |x|
    t = n * x
    ans << [ary[0..(t - 1)], ary[t...(t + n)]]
  end
  ans
end

ar = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
p generate(ar, 2)

p generate(ar, 1)
p generate(ar, 3)
p generate(ar, 4)

実行結果

$ ruby gen.rb
[[[1, 2], [3, 4]], [[1, 2, 3, 4], [5, 6]], [[1, 2, 3, 4, 5, 6], [7, 8]], [[1, 2, 3, 4, 5, 6, 7, 8], [9, 10]]]
[[[1], [2]], [[1, 2], [3]], [[1, 2, 3], [4]], [[1, 2, 3, 4], [5]], [[1, 2, 3, 4, 5], [6]], [[1, 2, 3, 4, 5, 6], [7]], [[1, 2, 3, 4, 5, 6, 7], [8]], [[1, 2, 3, 4, 5, 6, 7, 8], [9]], [[1, 2, 3, 4, 5, 6, 7, 8, 9], [10]]]
[[[1, 2, 3], [4, 5, 6]], [[1, 2, 3, 4, 5, 6], [7, 8, 9]]]
[[[1, 2, 3, 4], [5, 6, 7, 8]]]