时间戳

首页 > 解决方案 > 执行二叉树代码时出现 TypeError

python - 执行二叉树代码时出现 TypeError

问题描述

测试以下功能时出现错误。有人可以帮我吗？

代码：

class treenode(object):

def __init__(self, data, left=None, right=None):
    self.data = data
    self.left = left
    self.right = right
    
def largest_leaf_value(tnode):
    if tnode is None:
        return None
    res = tnode.data
    lres = largest_leaf_value(tnode.left)
    rres = largest_leaf_value(tnode.right)
    if lres > res:
        res = lres
    if rres > res:
        res = rres
    return res

这是测试脚本：

# test tree with 1 level i.e. root value only
input_tree = T.treenode(1)
expected = 3

result = a8q1.largest_leaf_value(input_tree)
assert result is expected, "{}: copying empty tree returned unexpected result".format(test_item)

# test a longer tree
input_tree = T.treenode(1, T.treenode(2, T.treenode(3, T.treenode(4, T.treenode(5)))))
expected = 5

result = a8q1.largest_leaf_value(input_tree)
assert result is expected, "{}: copying empty tree returned unexpected result".format(test_item)

这是我得到的错误：

Traceback (most recent call last):
  File "D:/CMPT 145/Assignment 8/a8q1_testing.py", line 60, in <module>
    result = a8q1.largest_leaf_value(input_tree)
  File "D:\CMPT 145\Assignment 8\a8q1.py", line 44, in largest_leaf_value
    if lres > res:
TypeError: '>' not supported between instances of 'NoneType' and 'int'

请让我知道为什么会这样？

标签： pythonbinary-search-tree

解决方案

由于您largest_leaf_value将None在其递归基本案例中返回，您需要准备好lres或被分配rres给None他们。

类型错误发生在您比较lres或的行rres上res，并告诉您None无法将值与 进行比较res。

因此，您有两种可能性：要么避免在 whenlres或rresis执行该比较None，要么不要让您的函数 return None，而是返回 -infinity （因为该值小于所有其他有限数值）。

第一种方法的解决方案：

def largest_leaf_value(tnode):
    if tnode is None:
        return None
    res = tnode.data
    lres = largest_leaf_value(tnode.left)
    rres = largest_leaf_value(tnode.right)
    # Only compare when not None:
    if lres is not None and lres > res:
        res = lres
    if rres is not None and rres > res:
        res = rres
    return res

第二种方法的解决方案：

def largest_leaf_value(tnode):
    if tnode is None:
        # Don't return None, but -infinity
        return float("-inf")
    res = tnode.data
    lres = largest_leaf_value(tnode.left)
    rres = largest_leaf_value(tnode.right)
    if lres > res:
        res = lres
    if rres > res:
        res = rres
    return res

None请注意，当您在初始调用中将空树（即）作为参数传递时，这将表现不同。从数学上讲，没有定义空集合中的最大值是多少，因此它将取决于您期望在这种情况下发生什么。

最后，您可以使用max()使第二个版本更短一些：

def largest_leaf_value(tnode):
    return (float("-inf") if tnode is None else
        max(tnode.data, largest_leaf_value(tnode.left), largest_leaf_value(tnode.right))
    )

推荐阅读