配列内の計算について

Question

いつもお世話になっております

今回の質問は、題名の通り配列内の計算です。

arr1 = [3, 2, 4, 3, 3, 3, 3, 3, 4, 3, 4]#回数

arr2 = [2, 1, 1, 2, 2, 2, 3, 2, 1, 2, 1, 1, 1, 2, 1, 1, 2, 1, 3, 2, 1, 1, 2, 2, 1, 1, 1, 1, 3, 1, 1, 2, 1, 2, 2]#足す数

arr1の要素の回数分をarr2で足したいのですがどうすればよろしいですか？

※arr[0]は3なので、arr2の要素を0から3回計算する(arr2[0..2])。次に、arr[1]は2なので、2回計算する。この時arr2はarr2[2]まで読み込んだので、次はarr2[3]から2回計算、つまりarr2[3..4]。こういう流れをarr1の要素数分やりたいのですがどのようにかけばうまくいきますか？

よろしくおねがいします

Answer 1

arr1 = [3, 2, 4, 3, 3, 3, 3, 3, 4, 3, 4]#回数

arr2 = [2, 1, 1, 2, 2, 2, 3, 2, 1, 2, 1, 1, 1, 2, 1, 1, 2, 1, 3, 2, 1, 1, 2, 2, 1, 1, 1, 1, 3, 1, 1, 2, 1, 2, 2]#足す数

##### 案1
n=0
sums = arr1.map do |x|
    n += x
    arr2[(n-x)...n].inject(&:+)
end
p sums

##### 案2
arr2dup = arr2.dup
sums = arr1.map do |x|
    arr2dup.shift(x).inject(&:+)
end
p sums

Comments

[probaseball]

ありがとうございます。

パターンがいろいろあるんですね、勉強になりました

Answer 2

arr1 = [3, 2, 4, 3, 3, 3, 3, 3, 4, 3, 4]#回数

arr2 = [2, 1, 1, 2, 2, 2, 3, 2, 1, 2, 1, 1, 1, 2, 1, 1, 2, 1, 3, 2, 1, 1, 2, 2, 1, 1, 1, 1, 3, 1, 1, 2, 1, 2, 2]

arr1.each{|x|
    print(arr2.slice!(0,x))
}