二叉搜索（排序）树的python实现_python二叉排序树的包

二叉搜索树依赖于这样的一个性质：小于父节点的节点都在左子树中，大于父节点的节点都在右子树中。

首先定义BST（BinarySearchTree）类和Node类：

class Node:
    def __init__(self,val):
        self.val=val
        self.lt=None
        self.rt=None
        self.parent=None

​
class BST:
    def __init__(self,node_list=None):
        if node_list==None:
            self.root=None
        else:
            self.root=Node(node_list[0])
            for i in node_list:
                self.insert(i)
 
​

初始化BST类时调用了insert函数，当插入一个节点到BST中时，首先比较插入节点与根节点的大小，如果插入节点小于根节点，则比较插入节点与左孩子大小，如果插入节点大于根节点，则比较插入节点与右孩子大小，一直循环下去直到寻找到插入位置。

非递归版：

    def insert(self,val):
        if not self.root:
            self.root=Node(val)
        else:
            new=Node(val)
            temp=self.root
            while temp:
                if val<=temp.val:
                    cur=temp
                    temp=temp.lt
                else:
                    cur=temp
                    temp=temp.rt
            new.parent=cur
            if val<=cur.val:
                cur.lt=new
            else:
                cur.rt=new

当搜索某个元素是否在BST中时，用到search函数。和insert函数一样，首先从根节点搜索，如果目标元素小于根节点，则查询左子树，否则查询右子树，直到搜索到目标元素。

非递归版：

    def search(self,obj):
        cur=self.rt
        while cur:
            if cur.val==obj:
                print('Find!')
                return
            if obj<cur.val:
                cur=cur.lt
            else:
                cur=cur.rt
        if not cur:
            print('No Find!')