二进制搜索树:插入操作

时间:2017-05-29 18:21:36

标签: python python-3.x binary-search-tree

我有以下代码实现BST。但是当我尝试通过调用插入函数插入元素时,它只打印出10和15.有人可以提供建议/更正吗?

class Node:
    def __init__(self,val):
        self.rightchild = None
        self.leftchild = None
        self.root = None
        self.value=val

    def insert(self,data):
        if self.value == data:
            return False

        elif self.value > data:
            if self.leftchild:
                return self.leftchild.insert(data)
            else:
                self.leftchild = Node(data)
                return True
        else:
            if self.rightchild:
                return self.rightchild.insert(data)
            else:
                self.rightchild = Node(data)
                return True

    def find(self,data):
        if self.value == data:
            return True
        elif self.value > data:
            if self.leftchild:
                return self.leftchild.find(data)
            else:
                return False
        else:
            if self.rightchild:
                return self.rightchild.find(data)
            else:
                return False

    def inorder(self):        
        if self.root:
            return self.leftchild.inorder()
        print(str(self.value))
        if self.rightchild:
            return self.rightchild.inorder()

class BST:    
    def __init__(self):
        self.root=None

    def insert(self,data):
        if self.root:
            return self.root.insert(data)
        else:
            self.root = Node(data)
            return True

    def find(self,data):
        if self.root:
            return self.root.find(data)
        else:
            return False

    def inorder(self):
        if self.root:
            self.root.inorder()

bst = BST()
bst.insert(10))
bst.insert(5)
bst.insert(15)
bst.insert(8)
bst.inorder()

1 个答案:

答案 0 :(得分:1)

Node#inorder中你有一个错字。第一行应该是

if self.leftchild:

另外,你过早地回来了。请勿在{{1​​}}中return,因为此方法是打印,而不是计算任何内容。改写为:

inorder

完成这些修复后,我得到了预期的答案:

def inorder(self):
   if self.leftchild:
       self.leftchild.inorder()
   print(str(self.value))
   if self.rightchild:
        self.rightchild.inorder()