如何在 Python 中使用不同类的实例调用类方法？

Question

我正在尝试使用两个类 - 节点和二叉树来实现二叉树。当我插入节点（左或右）时，我使用insert_left_node 和insert_right_node 方法，它们是class BinaryTree 的方法，但我也使用class Node 创建一个节点。每次插入节点后，返回当前对象。

现在，我如何使用返回的对象调用 BinaryTree 类的插入方法 - current。例如。在代码的倒数第二行，语句 n3 = n1.insert_left_node(33) 以 AttributeError: 'Node' object has no attribute 'insert_left_node'

失败

我需要另一种方法来实现这一点。

代码：

class Node(object):
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None


class BinaryTree(object):
    def __init__(self, root=None):
        self.root = Node(root)

    def insert_left_node(self, data):
        if not self.root:
            self.root = Node(data)
        else:
            current = self.root
            while True:
                if current.left:
                    current = current.left
                else:
                    current.left = Node(data)
                    break
            return current

    def insert_right_node(self, data):
        if not self.root:
            self.root = Node(data)
        else:
            current = self.root
            while True:
                if current.right:
                    current = current.right
                else:
                    current.right = Node(data)
                    break
            return current

if __name__ == '__main__':
    r = BinaryTree(34)  # root
    n1 = r.insert_left_node(22)
    n2 = r.insert_right_node(45)
    n3 = n1.insert_left_node(33)  # Fails
    print n3

Answer 1

您的请求实际上没有任何意义。要实现您想要的，您只需将所需的方法添加到您要使用的类中。尝试类似以下的操作：

class Node(object):
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None

    def insert_left_node(self, data):
        self.left = Node(data)

    def insert_right_node(self, data):
        self.right = Node(data)