数据结构与算法python—12.二叉搜索树及python实现与leetcode总结

一、什么是二叉搜索树？

  二叉查找树（Binary Search Tree)，(又:二叉搜索树，二叉排序树）它或者是一棵空树，或者是具有下列性质的二叉树:

1. 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值;
2. 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值;
3. 它的左、右子树也分别为二叉排序树。

二、二叉搜索树的实现

1.二叉搜索树的创建

节点创建：

class TreeNode:
    def __init__(self, key, value, left=None, right=None, parent=None):
        # 初始化key,value，左孩子、右孩子与父亲
        self.key = key
        self.value = value
        self.left_child = left
        self.right_child = right
        self.parent = parent
    # 判断是否有左孩子
    def has_left_child(self):
        return self.left_child
    # 判断是否有右孩子
    def has_right_child(self):
        return self.right_child
    # 判断是否是左孩子
    def is_left_child(self):
        return self.parent and self.left_child.parent == self
    # 判断是否是右孩子
    def is_right_child(self):
        return self.parent and self.right_child.parent == self
    # 判断是否是叶子节点
    def is_leaf(self):
        return not (self.left_child or self.right_child)
    # 判断是否是根节点
    def is_root(self):
        return not self.parent
    # 判断是否有其他子孩子
    def has_any_child(self):
        return self.right_child or self.right_child
    # 判断是否左孩子右孩子都存在
    def has_both_child(self):
        return self.left_child and self.right_child
    # 将初始化的值重新赋值
    def replace_data(self, key, value, lc, rc):
        self.key = key
        self.value = value
        self.left_child = lc
        self.right_child = rc

        if self.has_left_child():
            self.left_child.parent = self
            
        if self.has_right_child():
            self.right_child.parent = self

二叉搜索树创建：

class BinarySearchTree:
    def __init__(self):
        self.root = None
        self.size = 0

    def length(self):
        return self.size

    def __len__(self):
        return self.length()

    def __iter__(self):
        return self.root.__iter__()

2.二叉搜索树添加元素

思路框架：

从树的根节点开始，通过搜索二叉树来比较新的键值和当前节点的键值，

• 如果新的键值小于当前节点，则搜索左子树。
• 如果新的关键大于当前节点，则搜索右子树。

当搜索不到左（或右）子树，我们在树中所处的位置就是设置新节点的位置。向树中添加一个节点，创建一个新的TreeNode对象并在这个点的上一个节点中插入这个对象。

    # 二叉搜索树添加新元素
    def put(self, key, value):
        # 判断根节点是否为空
        # 如果不为空，则递归添加新元素
        if self.root:
            self._put(key, value, self.root)
        # 如果为空，则创建树节点，并设置该树节点为根节点
        else:
            self.root = TreeNode(key, value)
        self.size = self.size +