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

如果左边是无

，你需要至少得到正确的大小+ 1

我认为你需要这样的东西：

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

我还没试过。

Answer 2

第二个功能不准确，例如

x = Tree()
x.left = Tree()
x.right = None

在上面的示例中，x.size()将评估为零，因为您正在考虑left is None或right is None返回0（对于上面的示例，这很简单）。你需要调整你的逻辑。

def size(self):
    total = 1 #any instantiated object has a size of at least 1.
    if self.left is not None: #feel free to add additional validity checks, i.e. instanceof(Tree...)
        total += self.left.size()
    if self.right is not None:
        total += self.right.size()
    return total

Python中的二进制树大小函数

2 个答案: