整数を要素とする2次元配列が与えられたとき，その要素をすべて足し合わせた結果を出力するメソッドを作成したい。

Question

前提・実現したいこと

rubyで整数を要素とする2次元配列 mat が与えられたとき，その要素をすべて足し合わせた結果を出力するメソッドsum(mat)を作成したいのですが、配列をしたのように設定したらエラーが起きました。どこがダメなのかご教授ください。

発生している問題・エラーメッセージ

sum [[1, 3, 5, 7]]
TypeError: Array can't be coerced into Fixnum

該当のソースコード

def sum(mat)
 max = 0
 aaa(mat).each_index do |i|
  max +=  aaa(mat)[i]
end
end

def aaa(arr)
  if arr.size == 1
    arr
  elsif arr.size == 2
    bbb(arr[0],arr[1])
  else
    bbb(arr[0], aaa(arr[1..-1]))
  end
end


def bbb(arr1,arr2)
  ret = Array.new(arr1.size, 0)
  arr1.size.times{|n| ret[n] += arr1[n] + arr2[n]}
  ret
end

試したこと

irb> sum [[1, 2], [3, 4]]
=> 10
irb> sum [[-1, 0, 1], [-2, 0, 1]]
=> -1

上のような配列を試したところ正常に計算してくれました。

Answer 1

aaa() 関数に puts を入れてみると判るかと思いますが、入力が [[1, 3, 5, 7]] の場合は配列のサイズが 1 ですので、そのままの値で戻される事になります。そして、max += aaa(mat)[i] の aaa(mat)[i] は max += [1, 3, 5, 7] になるので当該のエラーが発生することになります。

def sum(mat)
  max = 0
  aaa(mat).each_index do |i|
    max += aaa(mat)[i]
  end
  return max
end

def aaa(arr)
  puts "aaa(arr) = #{arr} #{arr.size}"
  if arr.size == 1
    arr
          :

p sum [[1, 3, 5, 7]]

#
aaa(arr) = [[1, 3, 5, 7]] 1
aaa(arr) = [[1, 3, 5, 7]] 1
          :

まぁ、以下の様にしても良いかな、とは思います。

def sum(mat)
  return mat.flatten.sum
end

Comments

[hika2O21]

質問に丁寧に答えて下さり、ありがとうございます。
これからの勉強の参考させていただきます。

Answer 2

以下のようにしてください。

def sum(mat)
  max = 0
  aaa(mat).each_index do |i|
    max +=  aaa(mat)[i]
  end
  max
end

def aaa(arr)
  if arr.size == 1
    arr[0]
  elsif arr.size == 2
    bbb(arr[0],arr[1])
  else
    bbb(arr[0], aaa(arr[1..-1]))
  end
end


def bbb(arr1,arr2)
  ret = Array.new(arr1.size, 0)
  arr1.size.times{|n| ret[n] += arr1[n] + arr2[n]}
  ret
end
p sum [[1, 3, 5, 7]]

実行結果
16

Comments

[hika2O21]

質問に答えて下さり、ありがとうございます！
まだまだ未熟なところが多いので、参考にさせていただきます。