C++实现树 - 03 二叉排序树
数据结构与算法专栏 —— C++实现
写在前面：
上一讲我们介绍了二叉树的代码实现以及其遍历方法，这一讲我们来看一看进阶版的二叉树 —— 二叉排序树（二叉搜索树）。第一次接触这一讲知识点的朋友一时反应不过来不用太焦虑，其实是非常正常的，看讲解以及代码的同时动手画一画或者实现以下会让你更容易理解~
性质
二叉排序树它和普通的二叉树不同：
（1）如果根结点的左子树不为空，那么它左子树的所有结点的值都小于根结点。
（2）如果根结点的右子树不为空，那么它右子树的所有结点的值都大于根结点。
（3）根结点的左子树和右子树同样遵循上面两条性质。
根据性质我们可以知道，这颗排序二叉树的实现必然逃不了递归了，接下来我们来看看排序二叉树的代码是如何实现的。
二叉排序树的构建及插入
假设我们要构建起下面这颗二叉排序树，要进行如何操作呢：
（1）构建根结点 8 。
（2）插入结点 2 ，可以发现比根结点小，故插入根结点的左子树。
（3）插入结点 9 ，可以发现比根结点大，故插入根结点的右子树。
（4）插入结点 11 ，可以发现比根结点右孩子的值还要大，所以就插入到根结点右孩子的右子树。
（5）插入结点 23 ，可以发现比之前的值都要大，所以直接插到最右边的子树。
（6）插入结点 1 ，可以发现比之前的值都要小，所以直接插入到最左子树。
（7）最后插入结点 4 ，可以发现比根结点的左孩子要大，并且其左孩子没有右子树，所以直接插入即可。
整个操作的思想就是递归的思想，不断判断要插入结点的值和当前结点的值是大还是小，然后进入对应分支。
struct tree_node {
int data;
tree_node* left_child = NULL; //左孩子
tree_node* right_child = NULL; //右孩子
};
//插入孩子
void insert_child(tree_node*& root, int data) {
if (!root) {
tree_node* new_node = new tree_node();
new_node->data = data;
root = new_node;
}
else if (data < root->data)
insert_child(root->left_child, data);
else
insert_child(root->right_child, data);
}
二叉排序树的查找操作
我们这里的查找操作就不想数组查找那样那么直接，需要从根结点开始往下遍历进行比较。那么如果我们想要查找上面这颗二叉树的结点 4 ，该如何进行查找呢：
（1）先从根结点开始查找。
（2）可以发现我们要查找的结点 4 要比根结点 8 要小，所以我们进入根结点的左子树进行查找。
（3）结点 4 比结点 2 的值要大，所以我们进入其右子树进行查找。
（4）结果发现，当前遍历到的结点就是我们要查找的值，直接返回即可。
二叉排序树的删除操作
二叉排序树的删除操作就要比插入操作更加繁琐一些，需要分情况进行讨论，第一次学习不要慌张，一时头脑混乱非常正常，静下心来反复思考就可以吸收啦~
我们可以将删除操作分为三种情况：
（1）删除结点是叶子结点。有一个结点了。否则，同样delete删除结点即可，因为
假如我们要删除上面那棵树的结点 23 ，我们只用找到其父结点，然后将其父结点指向删除结点的指针指向空即可。
（2）如果删除的结点只有一个孩子
如果删除结点只有左孩子，那么只需将删除结点的左孩子连接到其父结点即可；如果删除结点只有右孩子，那么只需将删除结点的右孩子连接到其父结点即可。
如果删除结点是根结点则需另外操作，将根结点指针指向它的孩子即可。
假如我们要删除结点 11 ，那只用将结点 11 的右孩子直接接到其父结点 9 即可。
（3）如果删除结点既有左孩子也有右孩子。
这里我们有两种方法去实现，根结点和树中间的结点操作都是一样的，第一种方法是找到删除结点左子树中值最大的那个结点，然后与删除结点的值替换，并删除该结点。实际操作时，我们要从删除结点的左孩子开始查找，因为删除结点的左子树中，值最大的无非就是其左孩子或在左孩子的右子树当中，这里我们就拿删除根结点进行举例：
第二种方法是找到删除结点右孩子中最小的那个，交换并删除。同样，实际操作中也是从删除结点的右孩子开始查找，删除结点右子树中最小值一定在其右孩子或右孩子的左子树当中。这两种方法都保证删除后的树中结点仍然满足二叉排序树的性质，即左孩子都比根结点小，右孩子都比根结点大：
接下来我们来看代码的操作，这里我们是以上面第一种方法来实现的。另外，我们在实现代码时，会将一些操作合并在一起，因为有些情况的实现方法是相同的，比如说在当删除结点是叶子结点或删除结点只有一个孩子时，可以合在一起讨论，具体看下面的代码详解（代码略微有点长，需要大家细细品味和思考，这里删除操作如果找不到结点，就不会进行任何操作）：
//删除结点
void delete_node(tree_node*& root, int key) {
tree_node* cur = root; //当前指针
tree_node* fa = NULL; //父指针
//找到删除结点及其父结点
while (cur != NULL)
{
if (cur->data == key)
break;
fa = cur; //更新父指针
if (key < cur->data)
cur = cur->left_child;
else
cur = cur->right_child;
}
//如果根结点为空或没有该值，直接返回
if (!cur)
return;
//1、删除结点有左右孩子
if (cur->left_child && cur->right_child)
{
tree_node* temp = cur; //用于指向左子树最大值的结点的父结点
tree_node* left_max = cur->left_child; //指向左子树中的最大值
//找到删除结点左子树的最大值
while (left_max->right_child != NULL)
{
temp = left_max; //更新父指针
left_max = left_max->right_child;
}
cur->data = left_max->data; //将删除结点的前驱的值覆盖删除结点的值
//如果删除结点的左孩子没有右孩子
if (temp == cur)
{
temp->left_child = left_max->left_child;
delete[] left_max;
}
//如果删除结点的左孩子有右孩子
else
{
temp->right_child = left_max->left_child;
delete[] left_max;
}
}
//2、如果删除结点只有左孩子
else if (cur->left_child)
{
//如果删除结点是根结点
if (!fa)
root = root->left_child;
//如果删除结点的父结点的左孩子是删除结点
else if (fa->left_child == cur)
fa->left_child = cur->left_child;
//如果删除结点的父结点的右孩子是删除结点
else
fa->right_child = cur->left_child;
delete[] cur;
}
//3、如果删除结点只有右孩子
else if (cur->right_child)
{
//如果删除结点是根结点
if (!fa)
root = root->right_child;
//如果删除结点的父结点的左孩子是删除结点
else if (fa->left_child == cur)
fa->left_child = cur->right_child;
//如果删除结点的父结点的右孩子是删除结点
else
fa->right_child = cur->right_child;
delete[] cur;
}
//4、如果删除结点没有左右孩子即是叶子结点
else
{
//如果删除结点是根结点
if (!fa)
root = NULL;
//如果删除结点的父结点的左孩子是删除结点
else if (fa->left_child == cur)
fa->left_child = NULL;
//如果删除结点的父结点的右孩子是删除结点
else
fa->right_child = NULL;
delete[] cur;
}
}
全部代码
#include 如果大家有什么问题的话，欢迎在下方评论区进行讨论哦~
【上一讲】树 - 02 二叉树
【下一讲】树 - 04 二叉树的构建（数组）