红宝石，二叉树大小

Question

以下是我的代码。我想找到二叉树的大小。到目前为止，代码输出一个随机数。

def size
    size=1
    size +=@lchild.size unless @lchild.nil?
    size +=@rchild.size unless @richild.nil?
    size
  end

Answer 1

试试这个。

def size(node)
  if node.nil?
    0
  else
    size(node.left) + 1 + size(node.right)
  end
end