重构反向波兰表示法(RPN)或后缀表示法的反馈

时间:2013-07-02 03:59:52

标签: ruby rpn postfix-notation

Dev Bootcamp的一个前期练习是RPN计算器。我做了它的工作,但想重构反馈。非常感谢任何和所有帮助使这个代码更清洁。

class RPNCalculator
  def evaluate(rpn)
    a = rpn.split(' ')
    array = a.inject([]) do |array, i|    
      if i =~ /\d+/ 
        array << i.to_i
      else
        b = array.pop(2)
        case 
          when i == "+" then array << b[0] + b[1]  
          when i == '-' then array << b[0] - b[1]  
          when i == '*' then array << b[0] * b[1]  
          when i == '/' then array << b[0] / b[1]  
        end
      end
    end
    p array.pop
  end
end

calc = RPNCalculator.new
calc.evaluate('1 2 +')   # => 3
calc.evaluate('2 5 *')   # => 10
calc.evaluate('50 20 -') # => 30
calc.evaluate('70 10 4 + 5 * -') # => 0  

2 个答案:

答案 0 :(得分:2)

class RPNCalculator
  def evaluate rpn
    array = rpn.split(" ").inject([]) do |array, i|    
      if i =~ /\d+/ 
        array << i.to_i
      else
        b = array.pop(2)
        array << b[0].send(i, b[1])
      end
    end
    p array.pop
  end
end

答案 1 :(得分:1)

我倾向于避免使用case..when来支持查找表。所以我将你的代码更改为:

class RPNCalculator
  def evaluate(rpn)
    a = rpn.split(' ')
    array = a.inject([]) do |array, i|    
      if i =~ /\d+/ 
        array << i.to_i
      else
        array << array.pop(2).reduce(op(i))
      end
    end
    p array.pop
  end

  private

  def op(char)
    {'+'=>:+, '-'=>:-, '/'=>:/, '*'=>:*}[char]
  end
end

我也不相信你应该只是弹出2个操作数。 “1 2 3 +”将是有效的RPN,评估为6.整个堆栈应该减少。这也避免了突变,这是一件好事,因为它遵循更具功能性的风格。

class RPNCalculator
  def evaluate(rpn)
    a = rpn.split(' ')
    array = a.inject([]) do |array, i|    
      if i =~ /\d+/ 
        [*array, i.to_i]
      else
        [array.reduce(op(i))]
      end
    end
    p array.pop
  end

  private

  def op(char)
    {'+'=>:+, '-'=>:-, '/'=>:/, '*'=>:*}[char]
  end
end

我也使用[*arr, value]而不是实际修改数组来删除其他突变。

最后,我要避免直接从您的#evaluate方法打印,只返回数字。我也(再次)避免突变:

class RPNCalculator
  def evaluate(rpn)
    a = rpn.split(' ')
    stack = a.inject([]) do |stack, i|    
      if i =~ /\d+/ 
        [*stack, i.to_i]
      else
        [stack.reduce(op(i))]
      end
    end
    stack.last
  end

  private

  def op(char)
    {'+'=>:+, '-'=>:-, '/'=>:/, '*'=>:*}[char]
  end
end

我将'array'重命名为'stack',因为它是一个解析器堆栈,并且不仅仅是array