之前的博客中提到过,我学习采用的参考书是《数据结构与算法分析——C语言描述》。这门书的组织安排与国内广泛实用的教材《数据结构——C语言版》比较不同。这本书描述了一些树和二叉树的概念,举例讲解了什么是树的三种遍历之后,就开始重点讲解二叉查找树、平衡二叉树、AVL树、伸展树、B数了。这一篇博客,重点学习二叉查找树的概念和基本操作。
大家都知道,树的定义本身就带有递归性。因此,树的很多操作都涉及到了递归。二叉查找树的定义如下:
1.二叉查找树首先是一棵二叉树;
2.二叉查找树除了是二叉树外,还具有以下性质:对于树中的任何一个节点X,其左子树中的所有节点的关键字均小于X的关键字的值;而其右子树中的所有关键字的值均大于X的关键字的值。下面两棵二叉树中,左边的二叉树是二叉查找树而右边的不是。
二叉查找树的数据结构定义如下:
typedef int ElementType;
typedef struct TreeNode *Position;
typedef Position BiSearchTree;
struct TreeNode
{
ElementType Element;
BiSearchTree Left,Right;
};
二叉查找树的常用操作如下:
BiSearchTree MakeEmpty(BiSearchTree Tree);
Position Find(ElementType X, BiSearchTree T);
Position FindMin(BiSearchTree T);
Position FindMax(BiSearchTree T);
BiSearchTree Insert(ElementType X,BiSearchTree T);
BiSearchTree Delete(ElementType X, BiSearchTree T);
二叉排序树的操作与之前学习的一些数据结构的操作相比,可能难理解一些。下面我们挑比较难的几个一一讲解。
1.二叉查找树中查找指定的元素Find
二叉排序树的性质是二叉排序树很多操作的基本依据。既然要查找指定元素X,先比较X与根节点元素的关系,如果刚好相等,直接返回啦;如果X小于根节点的元素值,那么去根节点的左子树中查找,也就是调用Find方法,只是传递的参数是X和根节点的左子树;如果X大于根节点的元素值,那么去根节点的右子树中查找,也就是调用Find函数,只是传递的参数是X和根节点的右子树。
2.查找最小元素FindMin和最大元素FindMax
这两个操作是类似的。采用迭代或者递归都可以实现,查找最小元素只需要沿着左子树访问下去,查找最大元素则相反。下面我们会分别采用递归和迭代实现这两个操作。
3.插入操作Insert
插入操作其实类似于查找操作,插入过过程其实就是先得找到一个合适的位置。插入其实有下面几个情况:
(1)如果函数传进来的是空树,那么创建一棵树,将其元素值设置为X。这种情况显而易见;
(2)如果不是空树,那就比较根节点元素值和X的大小,如果X的值小于根节点的元素值,而此时的根节点的左子树为空,那么根节点的左孩子就是X元素的归宿啦;同样的道理,如果X的值大于根节点的元素值,而此时根节点的右子树为空,那么根节点的右孩子就是X元素的归宿啦。把握住这一点其实基本上就能把握住这个操作了。文字描述可能比较抽象,下面看图:
左边的图,如果要插入5,沿着树一直找到节点4,这时5>4并且4的右孩子为空,那么5就是4的右孩子。右边的图,要想插入8,沿着树找到9,发现8<9且9的左孩子为空,那么8就是9的左孩子。
当然,在实现这样一个过程的时候可以使用递归。
4.删除操作Delete
删除操作是我认为最复杂最不好理解的一个操作。如果没有仔细想明白整个过程,上来就看代码的话可能会很晕。删除操作分两步:第一步是查找,找的过程就涉及元素值之间的比较。我们重点说找到之后的操作。假设我们找到了这个节点,现在要删除,涉及三种情况:
(1)该节点是叶子。这还有什么好说的,直接删了一了百了;
(2)该节点只有一个孩子节点。也不复杂,让该节点的父节点直接指向其子节点就行了。当然,也别忘了回收该节点;
(3)该节点有两个孩子:找到该节点右子树中最小的节点,将其元素值赋给该节点,然后删除那个最小节点。这种情况看图:
说了半天了,下面看看完整的代码:
#include <stdio.h>
#include <stdlib.h>
typedef int ElementType;
typedef struct TreeNode *Position;
typedef Position BiSearchTree;
struct TreeNode
{
ElementType Element;
BiSearchTree Left,Right;
};
//建立一颗空树
BiSearchTree MakeEmpty(BiSearchTree Tree)
{
if(!Tree)
{
MakeEmpty(Tree->Left);
MakeEmpty(Tree->Right);
free(Tree);
}
return NULL;
}
//二叉查找树的Find操作
Position Find(ElementType X, BiSearchTree T)
{
if(T == NULL)
{
return NULL;
}
else
{
//关键字小于根节点的元素值
if(X < T->Element)
{
return Find(X,T->Left);
}
else if(X > T->Element)
{
return Find(X,T->Right);
}
else
{
return T;
}
}
}
//查找最小值:递归写法
Position FindMin(BiSearchTree T)
{
if(T == NULL)
{
return NULL;
}
else
{
if(T->Left == NULL)
{
return T;
}else
{
return FindMin(T->Left);
}
}
}
//查找最大值:非递归写法
Position FindMax(BiSearchTree T)
{
if(T->Right != NULL)
{
while(T->Right != NULL)
{
T = T->Right;
}
}
return T;
}
//插入元素X
BiSearchTree Insert(ElementType X,BiSearchTree T)
{
//当树为空树时
if(T == NULL)
{
T = malloc(sizeof(struct TreeNode));
if(T == NULL)
{
fprintf(stderr,"Out of Space!!!");
}
else
{
T->Element = X;
T->Left = NULL;
T->Right = NULL;
}
}
//树不为空时
else
{
if(X < T->Element)
{
T->Left = Insert(X,T->Left);
}
else if(X > T->Element)
{
T->Right = Insert(X,T->Right);
}
else
{
//do nothing!
}
}
return T;
}
//删除节点X
BiSearchTree Delete(ElementType X, BiSearchTree T)
{
Position TmpCell;
if(T==NULL)
{
fprintf(stderr,"Element does not exist!");
}
else if(X < T->Element)
{
T->Left = Delete(X,T->Left);
}
else if(X > T->Element)
{
T->Right = Delete(X,T->Right);
}
else if(T->Left && T->Right)
{
TmpCell = FindMin(T->Right);
T->Element = TmpCell->Element;
T->Right = Delete(T->Element,T->Right);
}
else
{
TmpCell = T;
if(T->Left == NULL)
{
T = T->Right;
}
else if(T->Right == NULL)
{
T = T->Left;
}
free(TmpCell);
}
return T;
}
int main(void)
{
BiSearchTree T;
int index;
int arr[10] = {10,9,8,7,6,1,2,3,4,5};
T = NULL;
for(index=0; index < 10; index++)
{
T = Insert(arr[index],T);
}
T = Insert(18,T);
T = Insert(15,T);
printf("The minimum element is %d\n",FindMin(T)->Element);
printf("The maxmium element is %d\n",FindMax(T)->Element);
return 0;
}