python - 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

我没试过