ETJava Beta | Java    注册   登录
  • 搜索:
  • BST 二叉搜索树 BinarySearchTree C++实现(递归/非递归)

    发表于      阅读(1)     博客类别:Crawler     转自:https://www.cnblogs.com/DSCL-ing/p/18366199
    如有侵权 请联系我们删除  (页面底部联系我们)  

    二叉搜索树

    基本概念

    二叉搜索树(BST,Binary Search Tree)

    二叉搜索树又称二叉排序树,它或者是一棵空树,或者是具有以下性质的二叉树:

    • 若它的左子树不为空,则左子树上所有节点的值都小于根节点的值
    • 若它的右子树不为空,则右子树上所有节点的值都大于根节点的值
    • 它的左右子树也分别为二叉搜索树

    image-20240818175416927

    int a[] = {8, 3, 1, 10, 6, 4, 7, 14, 13};

    二叉搜索树/二叉查找树也称二叉排序树,因为二叉排序树的中序遍历结果是升序

    常用结论

    二叉搜索树的左子树一定小于根,右子树一定大于根,结合定义递归子树可以得到

    • 左子树的最右节点是左子树的最大节点,右子树的最右节点是右子树的最大节点.

    • 左子树的最左节点是左子树的最小节点,右子树的最左节点是右子树的最小节点.

    • 二叉搜索树的最小节点是左子树的最左节点,最大节点是右子树的最右节点

    用途

    实际情况很少直接使用搜索二叉树,多是根据搜索二叉树的高效搜索特性,衍生出更为实用的高阶数据结构,例如平衡二叉搜索树(AVL树,红黑树)等...

    1. K模型:K模型即只有key作为关键码,结构中只需要存储Key即可,关键码即为需要搜索到的值。(在不在的问题)
      比如:给一个单词word,判断该单词是否拼写正确,具体方式如下:
      以词库中所有单词集合中的每个单词作为key,构建一棵二叉搜索树
      在二叉搜索树中检索该单词是否存在,存在则拼写正确,不存在则拼写错误。

    还有如:门禁系统,车库系统等...

    1. KV模型:每一个关键码key,都有与之对应的值Value,即<Key, Value>的键值对。该种方
      式在现实生活中非常常见: (通过一个值查找另一个值)
      比如英汉词典就是英文与中文的对应关系,通过英文可以快速找到与其对应的中文,英
      文单词与其对应的中文<word, chinese>就构成一种键值对;
      再比如统计单词次数,统计成功后,给定单词就可快速找到其出现的次数,单词与其出
      现次数就是<word, count>就构成一种键值对。

    还有如:通讯录

    二叉搜索树的性能分析

    插入和删除操作都必须先查找,查找效率代表了二叉搜索树中各个操作的性能。

    对有n个结点的二叉搜索树,若每个元素查找的概率相等,则二叉搜索树平均查找长度是结点在二叉搜索树的深度的函数,即结点越深,则比较次数越多。
    但对于同一个关键码集合,如果各关键码插入的次序不同,可能得到不同结构的二叉搜索树:

    image-20240820134152672

    最优情况下,二叉搜索树为完全二叉树(或者接近完全二叉树),其平均比较次数为:$log_2 N$ ($log_2 N$)
    最差情况下,二叉搜索树退化为单支树(或者类似单支),其平均比较次数为:$\frac{N}{2}$ ($\frac{N}{2}$)

    二叉搜索树的操作

    查找

    1. 从根开始比较,查找,比根大则往右边走查找,比根小则往左边走查找。
    2. 最多查找高度次,走到到空,还没找到,这个值不存在。

    插入

    1. 树为空,则直接新增节点,赋值给root指针(第一个节点就是root)
    2. 树不空,按二叉搜索树性质查找插入位置,插入新节点

    image-20240818181322297

    特别地

    • 同样一组数据,插入顺序不同,得到的二叉树也不同

    • 当插入的值已存在时,插入失败(不考虑multi)

    删除

    首先查找元素是否在二叉搜索树中,如果不存在,则返回.

    否则,根据树的结构定义,可以得到3种情况

    1. 要删除的结点无孩子结点
    2. 要删除的结点只有左孩子或右孩子时
    3. 要删除的结点有左、右孩子结点

    看起来有待删除节点有4中情况,实际情况:

    • 要删除的结点无孩子结点时,直接删除

    • 要删除的结点只有左孩子或右孩子时,将左孩子或右孩子给父亲

      1. 要删除的结点可能是父亲的左孩子或者是右孩子,有2*2种情况(要删除的结点是父亲的左孩子或右孩子)

      2. 左右孩子都是空时,也满足情况,因此可以合并无孩子结点情况

      3. 在1的前提下,恰好是根节点,也是一种情况(让另外一个孩子做根即可)

    • 要删除的结点有左右孩子(子树)时,需要找一个既要比左子树大也要比右子树小的节点来补上.

      根据递归定义得知,只有左孩子的最右结点和右孩子的最左结点符合条件,二选一即可

      当选择使用右孩子的最左结点时,有以下三种情况(与是不是根无关)

      1. 要删除的结点的右子树的最小结点恰好是要删除结点的右孩子.

      2. 要删除的结点的右子树的最小结点没有右孩子.

      3. 要删除的结点的右子树的最小结点有右孩子

        image-20240819214236004

        (上图举例分析)

    代码实现

    BSTree.hpp

    template<class K>
    struct BSTreeNode {
        BSTreeNode<K>* _left;
        BSTreeNode<K>* _right;
        K _key;
    
        BSTreeNode(K key) 
        :_key(key),_left(nullptr),_right(nullptr)
        {}
    };
    
    template<class K>
    class BSTree {
    public:
        using Node = BSTreeNode<K>;
        BSTree() = default;
        BSTree(const BSTree& bst) {
            _root = Copy(bst._root);
        }
        BSTree<K>& operator=(BSTree bst) { //拷贝复用
            swap(_root,bst.root);
            return *this;
        }
        ~BSTree() {
            Destroy(_root);
        }
    
    public:
        bool Insert(const K& key) {
            if (_root == nullptr) {
                _root = new Node(key);
                _root->_key = key;
                return true;
            }
            BSTreeNode<K>* cur = _root;
            BSTreeNode<K>* parent = _root;
            while (cur) {
                if (key < cur->_key) {
                    parent = cur;
                    cur = cur->_left;
                }
                else if (key > cur->_key) {
                    parent = cur;
                    cur = cur->_right;
                }
                else {
                    return false;
                }
            }
            //走出循环,说明树中不存在该节点, 可以插入
            cur = new BSTreeNode<K>(key);
            if (key < parent->_key) {
    
                parent->_left = cur;
            }
            else {
                parent->_right = cur;
            }
            return true;
        }
    
        bool Find(const K& key) {
            if (_root == nullptr) return false;
    
            Node* cur = _root;
            while (cur) {
                if (key < cur->_key) {
                    cur = cur->_left;
                }
                else if (key > cur->_key) {
                    cur = cur->_right;
                }
                else {
                    return true;
                }
            }
            // 从循环出来,说明没找着
            return false;
        }
    
        bool Erase(const K& key) {
            if (_root == nullptr) return false;
    
            Node* cur = _root;
            Node* parent = _root;
    
            while (cur) {
                if (key < cur->_key) {
                    parent = cur;
                    cur = cur->_left;
                }
                else if (key > cur->_key) {
                    parent = cur;
                    cur = cur->_right;
                }
                else {
                    //没有左孩子
                    if (cur->_left == nullptr) {
                        if (cur == _root) {
                            _root = cur->_right;
                        }
                        else if (parent->_left == cur) {
                            parent->_left = cur->_right;
                        }
                        else {
                            parent->_right = cur->_right;
                        }
                        delete cur;
                        return true;
                    }
                    //没有右孩子
                    else if (cur->_right == nullptr) {
                        if (cur == _root) {
                            _root = cur->_left;
                        }
                        if (parent->_left == cur) {
                            parent->_left = cur->_left;
                        }
                        else {
                            parent->_right = cur->_left;
                        }
                        delete cur;
                        return true;
                    }
                    //有左右孩子
                    else {
                        //找右孩子(子树)的最小结点/最左结点
                        Node* rightMin = cur->_right;  //明确不为空
                        Node* rightMinParent = cur;
                        while (rightMin->_left) {
                            rightMinParent = rightMin;
                            rightMin = rightMin->_left;
                        }
                        // 删除右子树最小结点有3种情况(与是不是根无关)
                        //1. 要删除的结点右子树最小结点恰好是自己的右孩子.
                        //2. 要删除的结点的右孩子的左子树的最左结点没有右孩子.
                        //3. 要删除的结点的右孩子的左子树的最左结点有右孩子.
                        //结论解析: 复用删除单结点代码,进行删除rightMin即可
                        K tmp = rightMin->_key;
                        Erase(rightMin->_key); //只能从根开始遍历,性能损失,但是二分查找很快,损失不大(理想情况,BST只学习用)
                        cur->_key = tmp;
                        return true;
                    } //有左右孩子的情况 
                } //找到了_继续处理的过程
            }//循环找的过程
            //循环结束,说明没找到
            return false;
        }//Erase [end]
    
        void InOrder() {
            _InOrder(_root);
            std::cout << std::endl;
        }
    
        bool InsertR(const K& key) {
            _InsertR(_root, key);
        }
    
        bool EraseR(const K& key) {
            return _EraseR(_root,key);
        }
    
    private:
    
        //此处返回值不能使用指针引用,虽然一定情况下可以使用(不推荐),至少目前不能引用空值.
        Node* Copy(const Node* root) {
            if (root == nullptr) {
                return nullptr;
            }
            Node* newRoot = new Node(root->_key);
            newRoot->_left = Copy(root->_left);
            newRoot->_right = Copy(root->_right);
            return newRoot;
        }
    
        //用不用引用无所谓,好习惯做到底
        //(析构子节点时,父节点两个成员会成为垂悬指针,但是接下来父亲也要析构了,指针变量也随之回收)
        void Destroy(Node*&root) {
            if (root == nullptr) {
                return ;
            }
            Destroy(root->_left);
            Destroy(root->_right);
    
            std::cout<<root->_key<<" ";
            delete root; //释放加自动置空
        }
    
        //练习递归+引用 -- 代码更加简洁
        bool _EraseR(Node*& root, const K&key) {
            //走到空,说明没找到,返回false
            if (root == nullptr) {
                return false;
            }
    
            //大于走右边,小于走左边
            if (key > root->_key) {
                return _EraseR(root->_right,key);
            }
            else if(key<root->_key) {
                return _EraseR(root->_left,key);
            }
            //找到了
            else {
                if (root->_left == nullptr) {
                    Node* del = root;
                    root = root->_right;
                    delete del;
                    return true;
                }
                else if (root->_right == nullptr) {
                    Node* del = root;
                    root = root->_left;
                    delete del;
                    return true;
                }
                //有左右孩子
                else {
                    Node* leftMax = root->_left;
                    //找左子树最大结点
                    while (leftMax->_right) {
                        leftMax = leftMax->_right;
                    }
                    std::swap(root->_key, leftMax->_key);
                    return _EraseR(root->_left, key);    //直接从左孩子开始递归删除.
                }
            }
    
        }
    
        //练习递归+引用指针的玩法,仅练习
        bool _InsertR(Node*& root, const K& key) { //引用的妙用,跨栈帧直接访问实参
            if (root == nullptr) {
                root == new Node(key);
                return true;
            }
            if (key == root->_key) return false;
            return (key > root->_key) ? _InsertR(root->_right, key) : _InsertR(root->_left, key);
        }
    
        void _InOrder(Node* root) {
            if (root == nullptr)  return;
            _InOrder(root->_left);
            std::cout << root->_key << " ";
            _InOrder(root->_right);
        }
    
    private:
        BSTreeNode<K>* _root = nullptr;
    };
    
    
    
    

    test.cc

    void test() {
        int a[] = { 8, 3, 1, 10, 6, 4, 7, 14, 13 };
        BSTree<int> bst;
        for (int i : a) {
            bst.Insert(i);
        }
        bst.InOrder();
    
        ////Find
        //std::cout << std::boolalpha << bst.Find(8) << std::endl; //true
        //std::cout << std::boolalpha << bst.Find(9) << std::endl; //false
    
        BSTree<int> cp(bst);
        cp.InOrder();
    
        //测试两孩子的三种情况即可
        bst.Erase(8);  //1. 要删除的结点的右子树的最小结点恰好是要删除结点的右孩子.
        bst.Erase(10); //2. 要删除的结点的右子树的最小结点没有右孩子
        bst.Insert(5); //构造有右孩子的最小结点
        bst.Erase(3);  //3. 要删除的结点的右子树的最小结点有右孩子
        bst.Erase(4);
        bst.Erase(7);
        bst.Erase(1);
        bst.Erase(14);
        bst.Erase(13);
        bst.Erase(6);
        bst.Erase(5);
        bst.InOrder();
    
        //禁止显式调用析构函数 --> 双重释放
        //bst.~BSTree();
        //cp.~BSTree();
        
    }
    
    int main() {
        test();
    }