rubyで二次元配列の比較をしたい

Question

ストラックアウトを作っています。
S=[["o", "x", "o"], ["o", "o", "x"], ["o", "x", "o"], ["x", "x", "x"]]
Z=[["1", "2", "3"], ["4", "5", "6"], ["7", "8", "9"], ["10", "11", "12"]]
で、Sで〇になった場合は、Zでその番号の数字の点数を加算していきたいです。

S.each do |one, two, three|

if one == "o"
puts Z.select {|(id,_ ,_)| id }
end

end

出力結果
1
2
3
4
5
6
7
8
9
10
11
12
1
2
3
4
5
6
7
8
9
10
11
12
1
2
3
4
5
6
7
8
9
10
11
12

です。何か良い方法はありませんでしょうか？よろしくお願いします。

Answer 1

それぞれ1次元にして(flatten)、両方をペアにして(transpose)、S側が"o"のものだけ選択してZ側をto_iしたものを(filter_map)、足す(sum)。

S=[["o", "x", "o"], ["o", "o", "x"], ["o", "x", "o"], ["x", "x", "x"]]
Z=[["1", "2", "3"], ["4", "5", "6"], ["7", "8", "9"], ["10", "11", "12"]]

score = [S,Z].map(&:flatten).transpose.filter_map{|ox,point| ox=="o" && point.to_i}.sum
p score

なお、filter_mapはRuby2.7以降の機能です。

Comments

[otn]

説明がよくわからなければ、
p([S,Z].map(&:flatten).transpose)
と、中間結果を見るとよいです。

[green66]

ありがとうございます。別の方法ですね！勉強になります。

[green66]

中間結果ですね。やってみます。

Answer 2

こういう事でしょうか？

num_sum = S.each_with_index.map{|r, i|
  r.each_with_index.map{|c, j|
    c == "o" ? Z[i][j].to_i : 0
  }
}.reduce(:+).sum

p num_sum
#=> 29

Comments

[green66]

回答ありがとうございます。今どの変数に合計値が格納されている状態でしょうか？

[melian]

回答を更新しました。 num_sum に格納されています。

[green66]

とてもわかりやすいです。ありがとうございました。

Answer 3

幾つかの方法を書いてみました。
最初は素直に書いています。そこから順々にコードを短くしていっています。

r.rb

# frozen_string_literal: true

def task_1(s, z)
  s.flatten.zip(z.flatten)     # [["o", "1"], ["x", "2"], ["o", "3"], ["o", "4"], ...
   .select { |x| x[0] == 'o' } # [["o", "1"], ["o", "3"], ["o", "4"], ...
   .map { |x| x[1].to_i }      # [1, 3, 4, 5, 7, 9]
   .sum # 29
end

def task_2(s, z)
  s.flatten.zip(z.flatten.map(&:to_i)) # [["o", 1], ["x", 2], ["o", 3], ["o", 4], ...
   .map { |x| x[0] == 'o' ? x[1] : 0 } # [1. 0, 3, 4, ...]
   .sum # 29
end

def task_3(s, z)
  s.flatten.zip(z.flatten)
   .each_with_object([]) { |x, mem| mem << x[1].to_i if x[0] == 'o' }
   .sum
end

def task_4(s, z)
  s.flatten.zip(z.flatten)
   .inject(0) { |mem, x| mem + (x[0] == 'o' ? x[1].to_i : 0) }
end

s = [%w[o x o], %w[o o x], %w[o x o], %w[x x x]]
z = [%w[1 2 3], %w[4 5 6], %w[7 8 9], %w[10 11 12]]

p task_1(s, z)
p task_2(s, z)
p task_3(s, z)
p task_4(s, z)

実行例

Comments

[green66]

回答ありがとうございます。回答が幾つかあると助かります。