目录
1.二叉搜索树的概念
二叉搜索树又称二叉排序树,它或者是一棵空树,或者是具有以下性质的二叉树:
1.若它的左子树不为空,则左子树上所有节点的值都小于根节点的值
2.若它的右子树不为空,则右子树上所有节点的值都大于根节点的值
3.它的左右子树也分别为二叉搜索树
2.二叉搜索树的实现
由上面的定义,我们知道二叉搜索树的左孩子是小于父亲的,而右孩子是大于父亲的。由这个规律我们就可以实现一下这个二叉搜索树。
2.1总体代码预览
template<class k> struct BSTreeNode { BSTreeNode<k>* _left; BSTreeNode<k>* _right; k _key; BSTreeNode(const k& key) :_left(nullptr) ,_right(nullptr) ,_key(key) {} }; template<class K> class BST { typedef BSTreeNode<K> Node; public: bool Insert(const K& key) { if (_root == nullptr) { _root = new Node(key); return true; } Node* parent = nullptr; Node* cur = _root; while (cur) { if (key > cur->_key) { parent = cur; cur = cur->_right; } else if (key < cur->_key) { parent = cur; cur = cur->_left; } else { return false; } } cur = new Node(key); if (key < parent->_key) { parent->_left = cur; } else { parent->_right = cur; } return true; } bool Find(const K& key) { Node* cur = _root; while (cur) { if (key > cur->_key) { cur = cur->_right; } else if (key < cur->_key) { cur = cur->_left; } else { return true; } } return false; } bool Erase(const K& key) { Node* parent = nullptr; Node* cur = _root; while (cur) { if (key > cur->_key) { parent = cur; cur = cur->_right; } else if (key < cur->_key) { parent = cur; cur = cur->_left; } else { //删除 if (cur->_left == nullptr) { if (cur == _root) { _root = cur->_right; } else { if (parent->_left == cur) { parent->_left = cur->_right; } else if (parent->_right == cur) { parent->_right = cur->_right; } } delete cur; } else if (cur->_right == nullptr) { if (cur == _root) { _root = cur->_left; } else { if (parent->_left == cur) { parent->_left = cur->_left; } else if (parent->_right == cur) { parent->_right = cur->_left; } } delete cur; } //cur左右都不为空 else { Node* rightminparent = cur; Node* rightmin = cur->_right; while (rightmin->_left) { rightminparent = rightmin; rightmin = rightmin->_left; } swap(rightmin->_key, cur->_key); if(rightminparent->_left== rightmin) rightminparent->_left = rightmin->_right; else rightminparent->_right = rightmin->_right; delete rightmin; } return true; } } return false; } void InOrder() { _InOrder(_root); } private: void _InOrder(Node* root) { if (root == nullptr) { return; } _InOrder(root->_left); cout << root->_key << endl; _InOrder(root->_right); } Node* _root=nullptr; };
2.2各个函数实现原理
链表结构体
template<class k> struct BSTreeNode { BSTreeNode<k>* _left; BSTreeNode<k>* _right; k _key; BSTreeNode(const k& key) :_left(nullptr) ,_right(nullptr) ,_key(key) {} };
既然是我们的二叉树,那么链表的部分肯定是不能少的吧,这个链表里面,我们来定义左右孩子的指针,还有用来进行标识的key值,接下来就可以写一下它的构造函数,这个节点一开始肯定是没有左右孩子的,所以我们给它们赋上空指针,而key就是我们传的key。
二叉搜索树的成员变量
Node* _root=nullptr;
二叉搜索树的成员变量就很简单了,我们直接定义一个结构体指针的变量的就可以了。Node类型是我们重命名链表名后的名字。
二叉搜索树的插入
bool Insert(const K& key) { if (_root == nullptr) { _root = new Node(key); return true; } Node* parent = nullptr; Node* cur = _root; while (cur) { if (key > cur->_key) { parent = cur; cur = cur->_right; } else if (key < cur->_key) { parent = cur; cur = cur->_left; } else { return false; } } cur = new Node(key); if (key < parent->_key) { parent->_left = cur; } else { parent->_right = cur; } return true; }
插入的逻辑其实非常简单,如果这棵树是一棵空树,那么我们直接创个根节点就可以了,不需要后续的操作,倘若不是空树,那我们就进行接下来的操作,我们先定义一个父亲节点parent和当前节点cur,然后按照我们的性质,key比cur中的大我们就往右,小的话我们就往左走,如果这个key在树中存在,我们就放回false,因为set是有去重功能的,就是说一棵树里不能存在两个一样的值。找到了适合的位置我们就出循环,然后开始创建节点,进行连接就可以了。
二叉搜索树的查找
bool Find(const K& key) { Node* cur = _root; while (cur) { if (key > cur->_key) { cur = cur->_right; } else if (key < cur->_key) { cur = cur->_left; } else { return true; } } return false; }
查找的功能就更简单了,我们只需要按照性质,左边比根小右边比根大就可以了。
二叉搜索树的遍历
void _InOrder(Node* root) { if (root == nullptr) { return; } _InOrder(root->_left); cout << root->_key << endl; _InOrder(root->_right); }
我们的二叉搜索树按照中序遍历是刚好是有序的,这个是我们二叉搜索树的意义之一,所以我们这里也是实现中序遍历。
中序遍历的思想就简单的多了,使用递归对它的各个节点进行遍历就可以了。
但是我们调用中序遍历肯定不想别人传个参数,所以我们再进行封装就可以了。
二叉搜索树的删除
bool Erase(const K& key) { Node* parent = nullptr; Node* cur = _root; while (cur) { if (key > cur->_key) { parent = cur; cur = cur->_right; } else if (key < cur->_key) { parent = cur; cur = cur->_left; } else { //删除 if (cur->_left == nullptr) { if (cur == _root) { _root = cur->_right; } else { if (parent->_left == cur) { parent->_left = cur->_right; } else if (parent->_right == cur) { parent->_right = cur->_right; } } delete cur; } else if (cur->_right == nullptr) { if (cur == _root) { _root = cur->_left; } else { if (parent->_left == cur) { parent->_left = cur->_left; } else if (parent->_right == cur) { parent->_right = cur->_left; } } delete cur; } //cur左右都不为空 else { Node* rightminparent = cur; Node* rightmin = cur->_right; while (rightmin->_left) { rightminparent = rightmin; rightmin = rightmin->_left; } swap(rightmin->_key, cur->_key); if(rightminparent->_left== rightmin) rightminparent->_left = rightmin->_right; else rightminparent->_right = rightmin->_right; delete rightmin; } return true; } } return false; }
二叉搜索树的删除才是我们的真正难点
我们最开始的操作还是一样的,我们要找那个被删除的节点,找到后就开始我们的操作了。我们有三种大情况:
第一种是要删除的节点左孩子为空的时候:
如图所示,我们的cur只有一边是有值的,这个时候我们先判断这个cur是父亲的左孩子还是右孩子,然后再把链表指向修改就行了,但是不要忘记了,我们的cur有可能是根节点,这个时候我们的父亲节点是空,这个时候是没法去给parent修改指向的,所以我们要单独的写一个判断。
第二种自然就是右孩子为空了,思路跟左孩子是一模一样的。
第三种就是我们的cur的左右孩子都不为空,这里的做法右两种,第一种是找cur的左子树的最大值,替换cur,然后再把cur给删除,第二种是找右子树的最小值,然后进行同样的操作,这两种做法都可以使其保持二叉搜索树该有的性质。这里我们选择第二种。
可以看到外面用循环来找到右子树的最小值,找到后交换值,我们没有直接让rightminparent的左节点直接接向rightmin的右子树是因为可能有这种情况
假如我们删的是8,这个时候右孩子是没有左节点的,这个时候我们就不可能rightminparent的左节点接向rightmin的右子树,所以我们进行了这个分类讨论,这个细节也是很难想到的。
至此我们的二叉搜索树的实现原理就讲完了,感谢大家的收看!
C++实现二叉搜索树(模型)
最新推荐文章于 2024-09-04 09:24:20 发布