数据结构——二叉搜索树的C++实现
GitHub下载链接:
https://github.com/Kyrie-leon/Data_Structures/tree/main/BSTreeNode
一、二叉搜索树的基本概念和存储结构
1.1 基本概念
二叉搜索树又称二叉排序树,它或者是一棵空树,或者是具有以下性质的二叉树:
- 若它的左子树不为空,则左子树上所有节点的值都小于根节点的值
- 若它的右子树不为空,则右子树上所有节点的值都大于根节点的值
- 它的左右子树也分别为二叉搜索树
1.2 二叉搜索树的结构
二叉搜索树采用链式结构存储
template<class K>
struct BSTNode {
K _key;
struct BSTNode<K> * _pleft; //左孩子
struct BSTNode<K> * _pright; //右孩子
BSTNode(const K& key)
:_key(key)
,_pleft(nullptr)
,_pright(nullptr)
{}
};
二、二叉搜索树的基本操作
2.1 二叉搜索树的查找
首先保证节点不为空的前提下:
- 如果根节点key==查找key,返回true
- 如果根节点key>查找key,遍历其左子树
- 如果根节点key<查找key,遍历其右子树
//二叉树的查找
const Node* Find(const K& key)
{
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
cur = cur->_pleft;
}
else if (cur->_key < key)
{
cur = cur->_pright;
}
else
{
return cur;
}
}
return nullptr; //空树直接返回
}
2.2 二叉搜索树的插入
//二叉搜索树插入
bool Insert(const K& key)
{
//1.空树直接插入
if (_root == nullptr)
{
_root = new Node(key);
return true;
}
//2.不为空
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
parent = cur;
cur = cur->_pleft;
}
else if (cur->_key < key)
{
parent = cur;
cur = cur->_pright;
}
else
{
return false;
}
}
Node* newNode = new Node(key);
if (parent->_key > key)
parent->_pleft = newNode;
else
parent->_pright = newNode;
return true;
}
2.3 二叉搜索树的删除
首先查找元素是否在二叉搜索树中,如果不存在,则返回, 否则要删除的结点可能分下面四种情况:
- a. 要删除的结点无孩子结点
- b. 要删除的结点只有左孩子结点
- c. 要删除的结点只有右孩子结点
- d. 要删除的结点有左、右孩子结点
看起来有待删除节点有4中情况,实际情况a可以与情况b或者c合并起来,因此真正的删除过程如下:
- 情况b:删除该结点且使被删除节点的双亲结点指向被删除节点的左孩子结点
- 情况c:删除该结点且使被删除节点的双亲结点指向被删除结点的右孩子结点
- 情况d:在它的右子树中寻找中序下的第一个结点(关键码最小),用它的值填补到被删除节点中,再来处理该结点的删除问题
//二叉搜索树的删除
bool Erase(const K& key)
{
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
parent = cur;
cur = cur->_pleft;
}
else if (cur->_key < key)
{
parent = cur;
cur = cur->_pright;
}
else
{
//三种情况:1.左为空 2.右为空 3.都不为空
if (cur->_pleft == nullptr)
{
//1.左为根节点,直接让右子树变为根节点
if (cur == _root)
{
_root = cur->_pright;
}
else
{
//两种情况:1.parent的左子树 2.parent的右子树
if (cur == parent->_pleft)
{
parent->_pleft = cur->_pright;
}
else
{
parent->_pright = cur->_pright;
}
}
delete cur;
}
else if (cur->_pright == nullptr)
{
//同上的情况
if (cur == _root)
{
_root = cur->_pleft;
}
else
{
//两种情况:1.parent的右子树 2.parent的左子树
if (cur == parent->_pright)
{
parent->_pright = cur->_pleft;
}
else
{
parent->_pleft = cur->_pleft;
}
}
delete cur;
}
else
{
//替代法,右树中找最小,需要注意一个情况右树就是最小值
Node* smParent = cur;
Node* subMin = cur->_pright;
//找右树中的最小值,一定是在右数的最左边
while (subMin->_pleft)
{
smParent = subMin;
subMin = subMin->_pleft;
}
//把要删除的节点值替换成右数最小值
cur->_key = subMin->_key;
if (smParent->_pleft == subMin)
{
smParent->_pleft = subMin->_pright;
}
else//右子树是最小值
{
smParent->_pright = subMin->_pright;
}
delete subMin;
}
return true;
}
}
return false;
}
三、二叉搜索树的实现
1.BSTree.hpp
#pragma once
template<class K>
struct BSTNode {
K _key;
struct BSTNode<K> * _pleft; //左孩子
struct BSTNode<K> * _pright; //右孩子
BSTNode(const K& key)
:_key(key)
,_pleft(nullptr)
,_pright(nullptr)
{}
};
template<class K>
class BSTree {
typedef struct BSTNode<K> Node;
public:
BSTree()
:_root(nullptr)
{}
//二叉树的查找
const Node* Find(const K& key)
{
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
cur = cur->_pleft;
}
else if (cur->_key < key)
{
cur = cur->_pright;
}
else
{
return cur;
}
}
return nullptr; //空树直接返回
}
//二叉搜索树插入
bool Insert(const K& key)
{
//1.空树直接插入
if (_root == nullptr)
{
_root = new Node(key);
return true;
}
//2.不为空
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
parent = cur;
cur = cur->_pleft;
}
else if (cur->_key < key)
{
parent = cur;
cur = cur->_pright;
}
else
{
return false;
}
}
Node* newNode = new Node(key);
if (parent->_key > key)
parent->_pleft = newNode;
else
parent->_pright = newNode;
return true;
}
//二叉搜索树的删除
bool Erase(const K& key)
{
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
parent = cur;
cur = cur->_pleft;
}
else if (cur->_key < key)
{
parent = cur;
cur = cur->_pright;
}
else
{
//三种情况:1.左为空 2.右为空 3.都不为空
if (cur->_pleft == nullptr)
{
//1.左为根节点,直接让右子树变为根节点
if (cur == _root)
{
_root = cur->_pright;
}
else
{
//两种情况:1.parent的左子树 2.parent的右子树
if (cur == parent->_pleft)
{
parent->_pleft = cur->_pright;
}
else
{
parent->_pright = cur->_pright;
}
}
delete cur;
}
else if (cur->_pright == nullptr)
{
//同上的情况
if (cur == _root)
{
_root = cur->_pleft;
}
else
{
//两种情况:1.parent的右子树 2.parent的左子树
if (cur == parent->_pright)
{
parent->_pright = cur->_pleft;
}
else
{
parent->_pleft = cur->_pleft;
}
}
delete cur;
}
else
{
//替代法,右树中找最小,需要注意一个情况右树就是最小值
Node* smParent = cur;
Node* subMin = cur->_pright;
//找右树中的最小值,一定是在右数的最左边
while (subMin->_pleft)
{
smParent = subMin;
subMin = subMin->_pleft;
}
//把要删除的节点值替换成右数最小值
cur->_key = subMin->_key;
if (smParent->_pleft == subMin)
{
smParent->_pleft = subMin->_pright;
}
else//右子树是最小值
{
smParent->_pright = subMin->_pright;
}
delete subMin;
}
return true;
}
}
return false;
}
//先序遍历
void _InOrder(Node* root)
{
if (root == nullptr)
return;
_InOrder(root->_pleft);
cout << root->_key << " ";
_InOrder(root->_pright);
}
void InOrder()
{
_InOrder(_root);
cout << endl;
}
private:
Node* _root;
};
四、二叉搜索树的应用——KV模型
- K模型:K模型即只有key作为关键码,结构中只需要存储Key即可,关键码即为需要搜索到的值。
比如:给一个单词word,判断该单词是否拼写正确,具体方式如下:
- 以单词集合中的每个单词作为key,构建一棵二叉搜索树
- 在二叉搜索树中检索该单词是否存在,存在则拼写正确,不存在则拼写错误。
- KV模型:每一个关键码key,都有与之对应的值Value,即<Key, Value>的键值对。该种方式在现实生活中非常常见:
比如英汉词典就是英文与中文的对应关系,通过英文可以快速找到与其对应的中文,英文单词与其对应的中文<word, chinese>就构成一种键值对;再比如统计单词次数,统计成功后,给定单词就可快速找到其出现的次数,单词与其出现次数就是<word, count>就构成一种键值对。
比如:实现一个简单的英汉词典dict,可以通过英文找到与其对应的中文,具体实现方式如下:
- <单词,中文含义>为键值对构造二叉搜索树,注意:二叉搜索树需要比较,键值对比较时只比较Key
- 查询英文单词时,只需给出英文单词,就可快速找到与其对应的key
1.BSTreeKV.hpp
#pragma once
template<class K, class V>
struct BSTNode {
const K _key; //不能改变
V _value;
struct BSTNode<K, V> * _pleft; //左孩子
struct BSTNode<K, V> * _pright; //右孩子
BSTNode(const K& key, const V& value)
:_key(key)
,_value(value)
, _pleft(nullptr)
, _pright(nullptr)
{}
};
template<class K, class V>
class BSTree {
typedef struct BSTNode<K, V> Node;
public:
BSTree()
:_root(nullptr)
{}
//二叉树的查找
Node* Find(const K& key)
{
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
cur = cur->_pleft;
}
else if (cur->_key < key)
{
cur = cur->_pright;
}
else
{
return cur;
}
}
return nullptr; //空树直接返回
}
//二叉搜索树插入
bool Insert(const K& key, const V& value)
{
//1.空树直接插入
if (_root == nullptr)
{
_root = new Node(key,value);
return true;
}
//2.不为空
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
parent = cur;
cur = cur->_pleft;
}
else if (cur->_key < key)
{
parent = cur;
cur = cur->_pright;
}
else
{
return false;
}
}
Node* newNode = new Node(key,value);
if (parent->_key > key)
parent->_pleft = newNode;
else
parent->_pright = newNode;
return true;
}
//二叉搜索树的删除
bool Erase(const K& key)
{
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_key > key)
{
parent = cur;
cur = cur->_pleft;
}
else if (cur->_key < key)
{
parent = cur;
cur = cur->_pright;
}
else
{
//三种情况:1.左为空 2.右为空 3.都不为空
if (cur->_pleft == nullptr)
{
//1.左为根节点,直接让右子树变为根节点
if (cur == _root)
{
_root = cur->_pright;
}
else
{
//两种情况:1.parent的左子树 2.parent的右子树
if (cur == parent->_pleft)
{
parent->_pleft = cur->_pright;
}
else
{
parent->_pright = cur->_pright;
}
}
delete cur;
}
else if (cur->_pright == nullptr)
{
//同上的情况
if (cur == _root)
{
_root = cur->_pleft;
}
else
{
//两种情况:1.parent的右子树 2.parent的左子树
if (cur == parent->_pright)
{
parent->_pright = cur->_pleft;
}
else
{
parent->_pleft = cur->_pleft;
}
}
delete cur;
}
else
{
//替代法,右树中找最小,需要注意一个情况右树就是最小值
Node* smParent = cur;
Node* subMin = cur->_pright;
//找右树中的最小值,一定是在右数的最左边
while (subMin->_pleft)
{
smParent = subMin;
subMin = subMin->_pleft;
}
//把要删除的节点值替换成右数最小值
cur->_key = subMin->_key;
if (smParent->_pleft == subMin)
{
smParent->_pleft = subMin->_pright;
}
else//右子树是最小值
{
smParent->_pright = subMin->_pright;
}
delete subMin;
}
return true;
}
}
return false;
}
//先序遍历
void _InOrder(Node* root)
{
if (root == nullptr)
return;
_InOrder(root->_pleft);
cout << root->_key << ":"<< root->_value<< endl;
_InOrder(root->_pright);
}
void InOrder()
{
_InOrder(_root);
cout << endl;
}
private:
Node* _root;
};