Python 中的二叉树大小函数

Question

我编写了几个函数来计算二叉树的大小。第一个（函数 1）工作得很好，并且是在类外部声明的，它不是类的成员函数。然而，第二个是类的成员函数给了我奇怪的结果。我很困惑！任何帮助，将不胜感激。

Function 1
    def size(root):
        if root is None:
            return 0
        else:
            return size(root.left)+ 1+ size(root.right)


Function 2
    def size(self):
        if self.left is None or self.right is None:
              return 0
        else:
              return self.left.size()+1+self.right.size()

Answer 1

if self.left is None or self.right is None:

如果其中之一为 None，则返回 0

如果 left 为 None，则至少需要获取 right + 1 的大小

我认为你需要类似的东西：

leftSize = self.left.size() if self.left else 0
rightSize = self.right.size() if self.right else 0
return leftSize + 1 + rightSize

我没试过。