重学数据结构007――二叉查找树

之前的博客中提到过，我学习采用的参考书是《数据结构与算法分析——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;
}