ruby とある問題での回答例。関数やクラス使えそうか？

Question

失礼します。
とある問題を説いていたら次のようなコードになりました。（問題の内容は拡散禁止されているため個人的な回答内容から読み取ってください。）

ちなみに入力される値はこれです。

～～

n
s_1
s_2
...
s_n

例：
3
SET 1 10
SET 2 20
ADD 40
↓
10 50
～～

x = gets.to_i
a=0
b=0
for i in 1..x do
    rd=gets.split(" ")
    if rd[0]=="SET"
        if rd[1].to_i==1
            a=rd[2].to_i
        elsif rd[1].to_i==2
            b=rd[2].to_i
        end
    elsif rd[0]=="ADD"
        b=a+rd[1].to_i
    elsif rd[0]=="SUB"
        b=a-rd[1].to_i
    end
end

puts "#{a} #{b}"

恐らく、これらはdef classなどをうまく使うこともできるのでは？とも思いましたが、あいにく使い慣れていないので頭で考えてみたら上記のような回答になりましたが、関数などを定義する場合はどのような記述になるでしょうか？

また、関数、クラスなどを使っていく際に必要な考え方、組み立て方、見通しの建て方などありましたらご指導よろしくお願いいたします。

Answer 1

さらに、クラスをつかってかいてみました。
メソッドへの分離をした時にあった swith-case の分岐をなくすこともしてみました。

class Mem
  def initialize(mem = [0] * 2)
    @mem = mem
    @actions = {
      SET: method(:do_set),
      ADD: method(:do_add),
      SUB: method(:do_sub)
    }
  end

  def dump
    @mem.join(' ')
  end

  def play(statements)
    statements.each do |statement|
      args = statement.split(' ')
      inst = args.shift.to_sym
      raise "bad instraction: #{inst}" unless @actions[inst]
      @actions[inst].call(args)
    end
  end

  private

  def do_set(args)
    @mem[args[0].to_i - 1] = args[1].to_i
  end

  def do_add(args)
    @mem[1] = @mem[0] + args[0].to_i
  end

  def do_sub(args)
    @mem[1] = @mem[0] - args[0].to_i
  end
end

# lines = []
# x = gets.to_i
# x.times { lines << gets }
lines = [
  'SET 1 10',
  'SET 2 20',
  'ADD 40'
]

mem = Mem.new [0, 0]
mem.play(lines)
puts mem.dump

Comments

[退会済みユーザー]

勉強になります。

Answer 2

関数をつかって処理を分割してみました。
gets でのデータ入力を処理から分離させることもしています。
各関数が mem を入力、mem を返すようになってます。

a.rb

def play(statements, mem)
  statements.each do |statement|
    tokens = statement.split(' ')
    inst = tokens.shift
    case inst
    when 'SET'
      mem = do_set(tokens, mem)
    when 'ADD'
      mem = do_add(tokens, mem)
    when 'SUB'
      mem = do_sub(tokens, mem)
    else
      raise "bad instraction: #{token[0]}"
    end
  end
  mem
end

def do_set(tokens, mem)
  pos = tokens[0].to_i - 1
  val = tokens[1].to_i
  mem[pos] = val
  mem
end

def do_add(tokens, mem)
  mem[1] = mem[0] + tokens[0].to_i
  mem
end

def do_sub(tokens, mem)
  mem[1] = mem[0] - tokens[0].to_i
  mem
end

lines = []
x = gets.to_i
x.times { lines << gets }

mem = [0, 0] # [a, b]
mem = play(lines, mem)
puts mem.join(' ')

Mem クラス(a と b の状態をもち、その状態を変更したり参照するメソッドを持つ) として定義すれば、各メソッドでいちいち mem を返すようなことは不要にできます。

クラスを定義すると、扱える変数を a, b だけでなくもっと多くするとか、SET, ADD, SUB 以外の操作もできるようにするといった場合に、変更とテストがしやすくなると思います。
(質問文のコード自体は短いので、そういった変更はいまのままでも簡単にできますが)

実際に、扱える変数を a..z に増やしたり、MUL, DIV, MOD といった処理をあつかえるように変更してみるとよいと思います。
(最終ゴールとしてはマシン語シュミレータをつくるとか、そのマシン語を生成するコンパイラーをつくるとか...)

Comments

[garchomp]

買いおつありがとうございます！
確かにこれだとメンテナンス性も高く、拡張なども非常にしやすいですね・・・

先のことも考えたプログラムを書けるように色々頑張ってみます！
マシン語・・・難しそうですね・・・頑張ってみます（汗）

Answer 3

arr = Array.new(2,0)
# p a,b
class Array
def order(x)
  num = x[1].to_i
  case x[0]
    when "SET"
      self[num - 1] = x[2].to_i
    when "ADD"
      self[1] = self[0] + num
    when "SUB"
      self[1] = self[0] - num
    end
  end
end

gets.to_i.times{ arr.order(gets.split) }

arr.join(" ").tap{|s| puts s }

自分がこの問題を解いた時の回答はこれですね
わざわざ新しいclassを書くよりはArrayクラスにメソッドを加えてしまった方が手取り早くコードが短くなると考えました。

Comments

[garchomp]

解答ありがとうございます！
arrayクラスにメソッドを加える…そんなこともできるんですね…
全然考え付きませんでした^^;

勉強り成りました！

Answer 4

a, b = 0, 0

x = gets.chomp.to_i

x.times do
  e = gets.chomp.split("\s").map { |t| t.to_i.to_s == t ? t.to_i : t }
  case e[0]
  when 'SET' then e[1] == 1 ? a = e[2] : b = e[2]
  when 'ADD' then b = a + e[1]
  when 'SUB' then b = a - e[1]
  end
end

puts "#{a} #{b}"

data = []

loop do
  input = gets
  break if input.nil?
  data << input.chomp
end

def answer(array)
  a, b = 0, 0

  array.drop(1).each do |e|
    e = e.split("\s").map { |t| t.to_i.to_s == t ? t.to_i : t }
    case e[0]
    when 'SET' then e[1] == 1 ? a = e[2] : b = e[2]
    when 'ADD' then b = a + e[1]
    when 'SUB' then b = a - e[1]
    end
  end

  return "#{a} #{b}"
end

puts answer(data)

質問者様のコードを参考にしてやってみました。
メソッド初心者の私の個人的な考えですが今回の場合は
メソッドを使わないほうが簡単なのではと思います。

Comments

[garchomp]

回答ありがとうございます！

メソッド使うかどうかは時と場合による。ですか。。。

同じ処理を複数回やる時にメソッドを考えた方が良さそうですね。

よく考えてみたら今回の場合ループで片付けられる。。。