Question

我想创建一个对象，让我们称其为模型，其中包括一个属性，二叉树的根。我希望该模型对象具有一种可构建特定深度的二叉树的方法。

我创建了一个类model，它使用树节点类binary_tree_node，如下所示：

class binary_tree_node:
    def __init__(self, data):
        self.data = data
        self.left_child = None
        self.right_child = None

class model:
    def __init__(self, max_depth = 3):
        self.root = None
        self.max_depth = max_depth

    def build_tree(self):
        self.build_tree_recursive_helper(self.root, 0)

    def build_tree_recursive_helper(self, node, current_depth):
        # create the new node
        node = binary_tree_node('data')

        # check base case
        if current_depth >= self.max_depth:
            return

        # make recursive calls
        self.build_tree_recursive_helper(node.left_child, current_depth + 1)
        self.build_tree_recursive_helper(node.right_child, current_depth + 1)

我希望能够实例化模型，构建树，然后对树进行自省，

m = model()
m.build_tree()
print(m.root.data)
>>> 'data'

但是相反，在尝试反省时，我得到了以下信息：

m = model()
m.build_tree()
print(m.root.data)

AttributeError                            Traceback (most recent call last)
<ipython-input-138-eaa2b3c07e85> in <module>
      1 m = model()
      2 m.build_tree()
----> 3 print(m.root.data)

AttributeError: 'NoneType' object has no attribute 'data'

这违反了我对python传递对象引用而不是值的理解。

如何修改binary_tree_node和model classes以达到预期的效果？

Answer 1

为什么不返回构建的节点并获取引用，如下所示：

max_depth = 4

def build_tree_recursive_helper(current_depth): 
    # check base case
    if current_depth >= max_depth:
        return None 

    node = binary_tree_node('data')

    # make recursive calls
    node.left_child = build_tree_recursive_helper(current_depth + 1)
    node.right_child = build_tree_recursive_helper(current_depth + 1)

    return node

请注意，您必须将self重新放入实现中。

建树时如何通过参考？

1 个答案: