重学数据结构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;}//插入元素XBiSearchTree 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;}//删除节点XBiSearchTree 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;}