Ruby 簡単な四則計算機

2つの変数が用意されていて、入力した命令に対して、以下の動作を行います。

• set i a : 変数 i に値 a(b) を代入する (変数a = 1, 変数b= 2)
• add a :「変数 a の値 + a」を計算し、計算結果を変数 b に代入する
• sub a :「変数 a の値 - a」を計算し、計算結果を変数 b に代入する
• mul a :「変数 a の値 * a」を計算し、計算結果を変数 b に代入する
• div a :「変数 a の値 / a」を計算し、計算結果を変数 b に代入する

初めの条件は以下の通りです。

1. 変数a = 0
2. 変数b = 0
3. 何回計算するか入力する。

入力例

何回計算しますか？  
3  
命令、値を入力してください  
set 1 10   
命令、値を入力してください  
set 2 20  
命令、値を入力してください  
add 40  

puts "何回計算しますか？"
n = gets.to_i
a = 0
b = 0
n.times do
  puts "命令、値を入力してください"
  s_i = gets.chomp.split.map(&:to_s)
  if s_i[0] == "set"
    i = s_i[1].to_i
    if i == 1
    a = s_i[2].to_i
    else i == 2
    b = s_i[2].to_i
    end
  elsif s_i[0] == "add"
    b = a + s_i[1].to_i
  elsif s_i[0] == "sub"
    b = a - s_i[1].to_i
  elsif s_i[0] == "mul"
    b = a * s_i[1].to_i
  else s_i[0] == "div"
    b = a / s_i[1].to_i
  end
end
puts "a=#{a}, b=#{b}"