二叉查找树

二叉查找树（BST：Binary Search Tree）是一种特殊的二叉树，它改善了二叉树节点查找的效率。二叉查找树有以下性质：

（1）若左子树不空，则左子树上所有节点的值均小于它的根节点的值

（2）若右子树不空，则右子树上所有节点的值均大于它的根节点的值

（3）左、右子树也分别为二叉排序树

（4）没有键值相等的节点

 

二叉查找树节点的定义：

typedef struct BSTreeNode
{
    int data;
    struct BSTreeNode *left;//左子树
    struct BSTreeNode *right;//右子树
}BSTree;

跟普通二叉树的节点定义相同

因为查找节点和插入节点都是在已经构建好二叉查找树的前提下才能进行的，在删除节点的时候才涉及到调整二叉树的操作，所以这里先以前序遍历的顺序直接输入一个二叉查找树，代码如下

/* 创建二叉查找树（数据以前序遍历顺序输入）*/
BSTree *Create_BSTreeNode(BSTree *nod)
{
    int num;

    scanf_s("%d", &num, 1);
    if (num == 0)    /* 假定输入的数据都为正数，以0作为NULL的标志 */
    {
        return NULL;
    }
    else
    {
        if ((nod = (BSTree *)malloc(sizeof(BSTree))) == NULL)
        {
            printf("内存空间不足");
            exit(0);
        }
        nod->data = num;
        nod->left = Create_BSTreeNode(nod->left);
        nod->right = Create_BSTreeNode(nod->right);

        return nod;
    }
}

/* 前序遍历二叉树，并打印 */
void PreOrder_Traverse(BSTree *nod, int level)
{
    if (nod == NULL)
    {
        return ;
    }

    printf("data = %d level = %d\n", nod->data, level);
    PreOrder_Traverse(nod->left, level + 1);
    PreOrder_Traverse(nod->right, level + 1);
}

 

1、查找节点（递归实现）

若根结点的关键字等于查找的关键字，查找成功，若小于根结点的关键字的值，递归查找左子树，若大于根结点的关键字的值，递归查找右子树，若子树为空，则查找失败，查找的操作较为简单，实现代码如下

/* 查找特定值 */
void SearchData(int targ, BSTree *nod)
{
    if (nod != NULL)
    {
        if (nod->data == targ)
        {
            printf("查找值存在，值为%d\n", nod->data);
        }
        else if (nod->data > targ)
        {
            SearchData(targ, nod->left);    //递归查找左子树
        }
        else if (nod->data < targ)
        {
            SearchData(targ, nod->right);    //递归查找右子树
        }
    }
    else if (nod == NULL)
    {
        printf("查找值不存在\n");
    }
}

通过 BST 查找节点，理想情况下我们需要检查的节点数可以减半。如下图中的 BST 树，包含了 15 个节点。从根节点开始执行查找算法，第一次比较决定我们是移向左子树还是右子树，对于任意一种情况，一旦执行这一步，我们需要访问的节点数就减少了一半，从 15 降到了 7。同样，下一步访问的节点也减少了一半，从 7 降到了 3，以此类推，根据这一特点，查找算法的时间复杂度应该是 O(log­₂n)

 

 

（图片来源：https://www.cnblogs.com/gaochundong/p/binary_search_tree.html）

 

对于 BST 查找算法来说，其十分依赖于树中节点的拓扑结构，也就是节点间的布局关系，当 BST 树中的节点以扇形结构散开时，对它的插入、删除和查找操作最优的情况下可以达到亚线性的运行时间 O(log₂n)，

因为当在 BST 中查找一个节点时，每一步比较操作后都会将节点的数量减少一半。尽管如此，如果拓扑结构像下图中的样子时，运行时间就会退减到线性时间 O(n)。因为每一步比较操作后还是需要逐个比较其余的节点，

也就是说，在这种情况下，在 BST 中查找节点与在数组（Array）中查找就基本类似了。

因此，BST 算法查找时间依赖于树的拓扑结构。最佳情况是 O(log­₂n)，而最坏情况是 O(n)

测试用例：

 

查找：以前序遍历顺序输入一个二叉查找树（0作为NULL标志）

 

 

2、插入节点（递归实现）

新插入的结点一定是一个新添加的叶子结点，如下图

 

虽然上面两种插入结果得到的二叉树都符合二叉查找树的性质，但是不满足“新插入的结点一定是一个新添加的叶子结点”，因为有了这个特点插入的操作变得相对简单，实现代码如下

/* 添加新节点 */
BSTree *AddNewNode(BSTree *cur, int NewData)
{
    if (cur == NULL)
    {
        if ((cur = (BSTree *)malloc(sizeof(BSTree))) == NULL)    //创建新节点
        {
            printf("内存不足");
            exit(0);
        }
        cur->data = NewData;
        cur->left = NULL;
        cur->right = NULL;

        return cur;
    }
    if (NewData > cur->data)
    {
        cur->right = AddNewNode(cur->right, NewData);
    }
    else if (NewData < cur->data)
    {
        cur->left = AddNewNode(cur->left, NewData);
    }
    else if (NewData == cur->data)
    {
        printf("不允许插入重复值\n");
        exit(0);
    }

    return cur;
}

 

测试用例：

 

插入：以前序遍历顺序输入一个二叉查找树

 

3、删除节点

删除节点的操作相对查找和插入要相对复杂一些，主要考虑以下三种情况（前两种情况操作较为简单，第三种较为复杂）

在删除操作前先用要找到待删除节点的位置（这里使用的递归，也可以改成迭代）

 

情形一：删除叶子节点

 

因为删除叶子节点不会破坏BST的结构，删除叶子节点的操作较为简单，步骤如下

1、判断待删除节点的左右子树是否为空，如果都为空那么就是叶子节点

2、判断待删除节点是待删除节点父节点的右子树还是左子树，将对应的指针赋值NULL

3、free待删除节点

 

实现代码：

/* 删除节点 */
void DeletNode(BSTree *parent, BSTree *cur, int DelData)
{
    BSTree *SNode = NULL; //后继节点
    BSTree *PSNode = NULL;    //后继节点的父节点

    if (DelData > cur->data)
    {
        DeletNode(cur, cur->right, DelData);
    }
    else if (DelData < cur->data)
    {
        DeletNode(cur, cur->left, DelData);
    }
    else if(DelData == cur->data)
    {
        if (cur->left == NULL && cur->right == NULL)    //删除节点为叶子节点
        {
            if (parent->left == cur)    //如果该节点是父节点的左子树
            {
                parent->left = NULL;
            }
            else if (parent->right == cur)    //如果该节点是父节点的右子树
            {
                parent->right = NULL;
            }
            free(cur);    //释放删除节点
        }

 

情形二：删除带有一个子节点的节点

（图片来源：https://www.cnblogs.com/songwenjie/p/8973217.html）

上图写了四种，但对待删除节点来说只有两种，只有左子树，或只有右子树，两种情况的处理方式基本相同，都是将待删除节点的左/右子树 赋值给 待删除节点的父节点的左/右子树

 

实现代码：

else if(cur->left != NULL && cur->right == NULL)    //待删除节点只有左子树
        {
            if (parent->left == cur)
            {
                parent->left = cur->left;
            }
            else if (parent->right == cur)
            {
                parent->right = cur->left;
            }
            free(cur);    //释放待删除节点
        }
        else if(cur->left == NULL && cur->right != NULL)    //待删除节点只有右子树
        {
            if (parent->left == cur)
            {
                parent->left = cur->right;
            }
            else if (parent->right == cur)
            {
                parent->right = cur->right;
            }
            free(cur);    //释放待删除节点
        }

 

情形三：删除带两个节点的节点

因为删除节点会有破坏 BST 正确结构的风险，删除带两个节点的节点操作显得较为复杂，首先需要找到待删除节点的 后继节点 和 该后继节点的父节点，（一个节点的后继节点是指，这个节点在中序遍历序列中的下一个节点，相应的，前驱节点是指这个节点在中序遍历序列中的上一个节点），删除节点的后继节点一定是删除节点右子树的最左侧节点，这篇随笔采用的方式是后继节点替代待删除节点的方式而不是前驱节点替代删除节点，需要考虑的情况如下

1、后继节点为待删除节点的子节点

在后继节点为待删除节点的子节点的前提下，该后继节点有右子树和没有右子树的操作是相同的，都是将 后继节点 替代 待删除节点，并将待删除节点的左子树 赋值给 后继节点的左子树

 

 实现代码：

else if(cur->left != NULL && cur->right != NULL)    //待删除节点既有左子树也有右子树
        {
            SNode = SearchSuccessorNode(cur->right);    //搜索后继节点
            PSNode = SearchParentofSNode(cur->right, cur->right);    //搜索后继节点的父节点

            if (cur->right == SNode)    //后继节点为待删除节点的右子树（后继节点有右子树和没有右子树的操作相同）
            {
                if (parent->left == cur)
                {
                    parent->left = SNode;
                    SNode->left = cur->left;

                    free(cur);
                }
                else if (parent->right == cur)
                {
                    parent->right = SNode;
                    SNode->left = cur->left;

                    free(cur);
                }
            }

 

2、后继节点不为待删除节点的子节点

这里后继节点还要在分为后继节点有子节点和没有子节点的情况

（1）后继节点没有右子节点

 

根据实现代码来标注上面的节点

删除后：

实现代码：

else if (cur->right != SNode && SNode->right == NULL)    //后继节点不为待删除节点的右子树，并且该后继节点没有右子树
            {
                if (parent->left == cur)
                {
                    parent->left = SNode;
                    SNode->left = cur->left;
                    SNode->right = cur->right;
                    PSNode->left = NULL;

                    free(cur);
                }
                else if (parent->right == cur)
                {
                    parent->right = SNode;
                    SNode->left = cur->left;
                    SNode->right = cur->right;
                    PSNode->left = NULL;

                    free(cur);
                }
            }

 

（2）后继节点有右子节点

 

删除后：

与上面的后继节点没有右子节点相比需要增加一个操作，需要将后继节点的右子树 赋值给 后继节点的父节点的左子树

实现代码： 

else if (cur->right != SNode && SNode->right != NULL)    //后继节点不为待删除节点的右子树，并且该后继节点有右子树
            {
                if (parent->left == cur)
                {
                    parent->left = SNode;
                    PSNode->left = SNode->right;    //后继节点的右子树作为后继节点父节点的左子树
                    SNode->left = cur->left;
                    SNode->right = cur->right;

                    free(cur);
                }
                else if (parent->right == cur)
                {
                    parent->right = SNode;
                    PSNode->left = SNode->right;    //后继节点的右子树作为后继节点父节点的左子树
                    SNode->left = cur->left;
                    SNode->right = cur->right;

                    free(cur);
                }
            }

 

测试数据：

一、“后继节点是删除节点的子节点”（因为后继节点有无子树的操作相同，这里只测试没有子树的情况）

 

二、“后继节点不是删除节点的子节点，且后继节点没有右子树”

 

 

三、“后继节点不是删除节点的子节点，且后继节点有右子树”

 

 

完整代码：（注：对free(cur)的位置进行了调整）

#include <stdio.h>
#include <stdlib.h>
#include <conio.h>

#define LEFT 1
#define RIGHT 2

typedef struct BSTreeNode
{
    int data;
    struct BSTreeNode *left;//左子树
    struct BSTreeNode *right;//右子树
}BSTree;

BSTree *Create_BSTreeNode(BSTree *nod);    //创建二叉查找树
void PreOrder_Traverse(BSTree *nod, int level);    //前序遍历二叉树，并打印
void SearchData(int targ, BSTree *nod);    //查找特定值
BSTree *AddNewNode(BSTree *cur, int NewData);    //添加新的节点
void DeletNode(BSTree *parent, BSTree *cur, int DelData);    //删除节点
BSTree *SearchSuccessorNode(BSTree *nod);    //搜索后继节点
BSTree *SearchParentofSNode(BSTree *Pnod, BSTree *nod);    //搜索后继节点的父节点

int main()
{
    BSTree *nod = NULL;
    //int num;
    //int key;
    int del;

    nod = Create_BSTreeNode(nod);
    PreOrder_Traverse(nod, 1);
    //printf("输出查找数据\n");
    //scanf_s("%d", &num, 1);
    //SearchData(num, nod);
    //printf("输出新插入数据\n");
    //scanf_s("%d", &key, 1);

    printf("输出删除的数据\n");
    scanf_s("%d", &del, 1);
    DeletNode(nod, nod, del);

    //AddNewNode(nod, key);
    PreOrder_Traverse(nod, 1);

    return 0;
}

/* 搜索后继节点的父节点 */
BSTree *SearchParentofSNode(BSTree *Pnod, BSTree *nod)
{
    while (1)
    {
        if (nod->left != NULL)
        {
            Pnod = nod;
            nod = nod->left;
        }
        else
        {
            break;
        }
    }

    return Pnod;
}

/* 搜索后继节点 */
BSTree *SearchSuccessorNode(BSTree *nod)
{
    while (1)
    {
        if (nod->left != NULL)
        {
            nod = nod->left;
        }
        else
        {
            break;
        }
    }

    return nod;
}

/* 删除节点 */
/* cur为待删除节点， parent为待删除节点的父节点 */
void DeletNode(BSTree *parent, BSTree *cur, int DelData)
{
    BSTree *SNode = NULL; //后继节点
    BSTree *PSNode = NULL;    //后继节点的父节点

    if (DelData > cur->data)
    {
        DeletNode(cur, cur->right, DelData);
    }
    else if (DelData < cur->data)
    {
        DeletNode(cur, cur->left, DelData);
    }
    else if(DelData == cur->data)
    {
        if (cur->left == NULL && cur->right == NULL)    //待删除节点为叶子节点
        {
            if (parent->left == cur)    //如果该节点是父节点的左子树
            {
                parent->left = NULL;
            }
            else if (parent->right == cur)    //如果该节点是父节点的右子树
            {
                parent->right = NULL;
            }
        }
        else if(cur->left != NULL && cur->right == NULL)    //待删除节点只有左子树
        {
            if (parent->left == cur)
            {
                parent->left = cur->left;
            }
            else if (parent->right == cur)
            {
                parent->right = cur->left;
            }
        }
        else if(cur->left == NULL && cur->right != NULL)    //待删除节点只有右子树
        {
            if (parent->left == cur)
            {
                parent->left = cur->right;
            }
            else if (parent->right == cur)
            {
                parent->right = cur->right;
            }
        }
        else if(cur->left != NULL && cur->right != NULL)    //待删除节点既有左子树也有右子树
        {
            SNode = SearchSuccessorNode(cur->right);    //搜索后继节点
            PSNode = SearchParentofSNode(cur->right, cur->right);    //搜索后继节点的父节点

            if (cur->right == SNode)    //后继节点为待删除节点的右子树（后继节点有右子树和没有右子树的操作相同）
            {
                if (parent->left == cur)
                {
                    parent->left = SNode;
                    SNode->left = cur->left;
                }
                else if (parent->right == cur)
                {
                    parent->right = SNode;
                    SNode->left = cur->left;
                }
            }
            else if (cur->right != SNode && SNode->right == NULL)    //后继节点不为待删除节点的右子树，并且该后继节点没有右子树
            {
                if (parent->left == cur)
                {
                    parent->left = SNode;
                    SNode->left = cur->left;
                    SNode->right = cur->right;
                    PSNode->left = NULL;
                }
                else if (parent->right == cur)
                {
                    parent->right = SNode;
                    SNode->left = cur->left;
                    SNode->right = cur->right;
                    PSNode->left = NULL;
                }
            }
            else if (cur->right != SNode && SNode->right != NULL)    //后继节点不为待删除节点的右子树，并且该后继节点有右子树
            {
                if (parent->left == cur)
                {
                    parent->left = SNode;
                    PSNode->left = SNode->right;    //后继节点的右子树作为后继节点父节点的左子树
                    SNode->left = cur->left;
                    SNode->right = cur->right;
                }
                else if (parent->right == cur)
                {
                    parent->right = SNode;
                    PSNode->left = SNode->right;    //后继节点的右子树作为后继节点父节点的左子树
                    SNode->left = cur->left;
                    SNode->right = cur->right;
                }
            }
        }
        free(cur);    //释放待删除节点
    }
}

/* 添加新节点 */
BSTree *AddNewNode(BSTree *cur, int NewData)
{
    if (cur == NULL)
    {
        if ((cur = (BSTree *)malloc(sizeof(BSTree))) == NULL)    //创建新节点
        {
            printf("内存不足");
            exit(0);
        }
        cur->data = NewData;
        cur->left = NULL;
        cur->right = NULL;

        return cur;
    }
    if (NewData > cur->data)
    {
        cur->right = AddNewNode(cur->right, NewData);
    }
    else if (NewData < cur->data)
    {
        cur->left = AddNewNode(cur->left, NewData);
    }
    else if (NewData == cur->data)
    {
        printf("不允许插入重复值\n");
        exit(0);
    }

    return cur;
}

/* 查找特定值 */
void SearchData(int targ, BSTree *nod)
{
    if (nod != NULL)
    {
        if (nod->data == targ)
        {
            printf("查找值存在，值为%d\n", nod->data);
        }
        else if (nod->data > targ)
        {
            SearchData(targ, nod->left);    //递归查找左子树
        }
        else if (nod->data < targ)
        {
            SearchData(targ, nod->right);    //递归查找右子树
        }
    }
    else if (nod == NULL)
    {
        printf("查找值不存在\n");
    }
}

/* 创建二叉查找树（数据以前序遍历顺序输入）*/
BSTree *Create_BSTreeNode(BSTree *nod)
{
    int num;

    scanf_s("%d", &num, 1);
    if (num == 0)    /* 假定输入的数据都为正数，以0作为NULL的标志 */
    {
        return NULL;
    }
    else
    {
        if ((nod = (BSTree *)malloc(sizeof(BSTree))) == NULL)
        {
            printf("内存空间不足");
            exit(0);
        }
        nod->data = num;
        nod->left = Create_BSTreeNode(nod->left);
        nod->right = Create_BSTreeNode(nod->right);

        return nod;
    }
}

/* 前序遍历二叉树，并打印 */
void PreOrder_Traverse(BSTree *nod, int level)
{
    if (nod == NULL)
    {
        return ;
    }

    printf("data = %d level = %d\n", nod->data, level);
    PreOrder_Traverse(nod->left, level + 1);
    PreOrder_Traverse(nod->right, level + 1);
}

因为主要是分析二叉查找树的查找、插入、删除操作，没有考虑二叉树为空 或 待删除节点为根节点的情况