二叉搜索树python实现——python——系列1

这几天讲了二叉搜索树的相关算法，好像教材没给出完整的实现过程，都是缺胳膊少腿，学生自学的过程中老碰到问题，所以今天抽空实现下插入算法，便于学生查阅，检查自己的实现。其它功能请大家自己补全，偶实在没时间一个一个再敲了。

1. 废话不多说，完整框架列一下：

# ###########################
# @author: Mr liu      #
# @time: 20190429 16:32     #
# ###########################
# BST树的节点类
class Node:
    def __init__(self, key):
        self.key = key
        self.left = None
        self.right = None
        self.parent = None
# BST树及其各种操作
class BST:
    def __init__(self, *args):
        self.Root = None
    
    def find(self, key, *args):
        #边界条件作用：不断找根节点赋值给start，为空时start=None
        if len(args) == 0: 
            start = self.Root
        else:
            start = args[0] 
        if not start: #如果没有节点，则返回None
            return None 
        #如果根节点存在，则比较key值，遵循BST的右大左小规律递归查询
        if key == start.key: #找到则返回该节点
            return start
        elif key > start.key: #大于向右转
            return self.find(key, start.right)
        else: #小于向左转
            return self.find(key, start.left)
    
    def insert(self, key, *args):
        if not self.Root: #树为空
            self.Root = Node(key) #当前节点作为根节点
        elif len(args) == 0: #查找当前节点是否存在,不存在则执行插入操作
            if not self.find(key, self.Root): #从根节点开始查，没找到返回None
                self.insert(key, self.Root) #既然没找到，那就插入操作吧
        else: #树不为空了，当前节点在BST中也不存在了，那就开始插入操作
            child = Node(key) #用节点类创建孩子节点
            parent = args[0]  #找到当前根节点，设置为双亲节点
            if child.key > parent.key: #比较key值，大于则向右转
                if not parent.right: #看看当前根节点的右有人不？没人？这孩子我要了
                    parent.right = child
                    child.parent = parent
                else: #有人？那以当前根节点的右孩子为根节点，递归吧，看看谁要这孩子
                    self.insert(key, parent.right)
            else: #小于则向左转
                if not parent.left:
                    parent.left = child
                    child.parent = parent
                else:
                    self.insert(key, parent.left)
                    
def main():
    tree = BST()
    A = [8, 3, 10, 1, 6, 14, 4, 7, 13]
    for i in A:
        tree.insert(i)

if __name__ == '__main__':
    main()                

2. 刚这玩意儿，看不到没用，所以修改库函数，因为改动了库函数，牵扯代码量太大，没法贴了，就看看效果吧。每个节点插入的时候都打印了一遍，所以出现了9个子图，按照BST的插入规则插入了节点。当然也可以自己设计一个打印函数，就是类似字符串的那种打印方式，忘记了？那还不赶快去数据结构里恶补下？