什么会使此 BST 代码中的插入方法起作用？

Question

除了插入方法外，该类似乎运行良好。所有打印语句方法都有效。我想在树中插入一个节点，但是一旦我调用r.insertion(7,3)，它就不起作用并说没有“键属性”。如何修复这个二叉搜索树？

class BinarySearchTree:
    def __init__(self,rootObj):
        self.key = rootObj
        self.leftChild = None
        self.rightChild = None

    def insertLeft(self,newNode):
        if self.leftChild == None:
            self.leftChild = BinarySearchTree(newNode)
        else:
            t = BinarySearchTree(newNode)
            t.leftChild = self.leftChild
            self.leftChild = t

    def insertRight(self,newNode):
        if self.rightChild == None:
            self.rightChild = BinarySearchTree(newNode)
        else:
            t = BinarySearchTree(newNode)
            t.rightChild = self.rightChild
            self.rightChild = t
            
    def insertion(self,root, num):
    
        if root == None:
            return Node(num)
        #check if less than
        
        if num < root.key:
            root.left = insertion(self, root.left, num)
            
        elif num >= root.key:
            root.right = insertion(self, root.right, num)
            
        return root


    def getRightChild(self):
        return self.rightChild

    def getLeftChild(self):
        return self.leftChild

    def setRootVal(self,obj):
        self.key = obj

    def getRootVal(self):
        return self.key


r = BinarySearchTree(7)
print(r.getRootVal())
print(r.getLeftChild())
r.insertLeft(4)
print(r.getLeftChild())
print(r.getLeftChild().getRootVal())
r.insertRight(9)
print(r.getRightChild())
print(r.getRightChild().getRootVal())
r.getRightChild().setRootVal('hello')
print(r.getRightChild().getRootVal())
print(r.insertion(7,3))

谢谢。

Answer 1

根据您的函数定义，def insertion(self,root, num) 采用 2 个主要参数：

1. root：根据您的实现，它必须是 BinarySearchTree 类的对象
2. num，这将是一个 int

当你调用r.insertion(7,3) 时，num 得到一个 int；但是，root 参数也得到了int，这是不正确的（这就是错误背后的原因）。如果你想插入r，你必须使用r.insertion(r, 3)来调用它，因为r是你想插入值的BST的根节点。