Ruby：计算二叉树的所有节点值的总和

Question

我有一个如下的二叉树实现。我想添加一个递归总结二叉树所有节点值的方法：

class BST

  class Node
    attr_reader :value, :left, :right

    def initialize(value)
      @value = value
      @left = nil
      @right = nil
    end

    def insert(value)
      if value <= @value
        @left.nil? ? @left = Node.new(value) : @left.insert(value)
      elsif value > @value
        @right.nil? ? @right = Node.new(value) : @right.insert(value)
      end
    end

  end

  def initialize
    @root = nil
  end

  def insert(value)
    @root.nil? ? @root = Node.new(value) : @root.insert(value)
  end

end

我找到了其他语言的答案，但不幸的是 Ruby 没有。

Answer 1

我认为您在 cmets 中的代码是：

def sum(node=@root)
  return if node.nil?
  total += node.value
  sum(node.left)
  sum(node.right)
end

这个想法几乎没问题。 nil 节点中没有求和；然后将当前节点的值、左节点和右节点相加。以下是错误：

• total += node.value 是我们第一次看到total。这会将其初始化为nil。当您尝试向其中添加 node.value 时，您会收到您描述的错误。为避免这种情况，total 必须已经存在，或者您可以分配 node.value 给它。

• 如果函数结束时没有执行return 语句，则返回最后一个计算的表达式；在这种情况下，sum(node.right)。 sum返回total不是更好吗？

• 相反，sum(node.left) 可能会做一些求和...但它的返回值被丢弃。将它添加到total 可能是有意义的。说到总数，也许我们应该对 sum(node.right) 做同样的事情。

• 最后，return if node.nil? 说您拒绝对实际上不是节点的节点求和。太好了...除了return 返回nil，如果您尝试将nil 与某物相加，则效果不佳。这里有两种解决方案：在输入之前拒绝对一个节点求和，或者说一个nil节点的值为0，这不会影响求和。

综合来看，这是我的两个版本：

# refuse to enter a nil node:
def sum(node=@root)
  total = node.value
  total += sum(node.left) unless node.left.nil?
  total += sum(node.right) unless node.right.nil?
  total
end

# treat nil nodes as if they were zeroes:
def sum(node=@root)
  return 0 if node.nil?
  node.value + sum(node.left) + sum(node.right)
end