二叉查找树的实现与应用

#include<stdio.h>
#include<stdlib.h>
#include<malloc.h>
#define OK 1
#define ERROR 0
#define TRUE 1
#define FALSE 0
#define OVERFLOW -1
#define N 5
//初始化.插入，删除，递归，中序遍历
typedef int Status;
typedef int RcdType;
typedef struct BSTNode{
    RcdType data;//数据元素
    struct BSTNode *lchild,*rchild;//左右孩子指针
}BSTNode,*BSTree;//二叉查找树

BSTree SearchBST(BSTree T,RcdType key);
BSTree InserBST(BSTree T,RcdType key);
BSTree DeleteBST(BSTree *T,RcdType key);
void DeleteNode(BSTree *p);
void InOderTraverse_LVR(BSTree T);

BSTree SearchBST(BSTree T,RcdType key){//二叉查找树递归查找
    if(NULL==T)return NULL;//查找失败
    if(T->data==key)return T;//查找成功
    if(T->data>key)
            return SearchBST(T->lchild,key);//在左子树上继续查找
    else if(T->data<key)
            return SearchBST(T->rchild,key);
    else
        return TRUE;
}
BSTree InserBST(BSTree T,RcdType key){
    if(T==NULL){
        T=(BSTree)malloc(sizeof(BSTNode));
        T->data=key;
        T->lchild=NULL;
        T->rchild=NULL;
        printf("%d插入树中\n",key);
        return T;
    }
    if(key<T->data)
      T->lchild=InserBST(T->lchild,key);
    else if(key>T->data)
      T->rchild=InserBST(T->rchild,key);
    else
        printf("%d已在树中，不能再次插入。\n",key);
    return T;
}


BSTree DeleteBST(BSTree *T,RcdType key){
//若二叉查找树T中存在为key的结点，则删除该结点，并返回TRUE；否则返回FALSE
    if(NULL==T)return FALSE;
    if(key==(*T)->data){//找到值为key的结点
        DeleteNode(&T);//删除值为key的结点
        return T;
    }
    else if((*T)->data>key)
        return DeleteBST((*T)->lchild,key);//对左子树递归并返回结果
            return DeleteBST((*T)->rchild,key);
}
void DeleteNode(BSTree *p){
//删除二叉查找树中的p结点，引用形参p的实参是要删除的p结点的双亲指向其的指针域
    BSTNode *q,*s;
    q=p;//令q指向要删除的p
    if(NULL==(*p)->rchild){//被删的右子树为空
       p=(*p)->lchild;//置p的左子树为p的结点的双亲结点相应的子树
       free(q);
    }else if(NULL==(*p)->lchild){//被删p结点的左子树为空
        p=(*p)->rchild;
       free(q);
    }else{//左右子树都为空
        s=(*p)->lchild;//s指向要删除的p的左孩子
        while(s->rchild!=NULL)//向右走到尽头，令s指向被删除点的直接前驱
            { q=s;s=s->rchild; }
            (*p)->data=s->data;//将直接前驱s结点的值赋予被删除的点
            if(q==p)//s结点是被删结点的左孩子
            q->lchild=s->lchild;//删除s结点只需将左子树置为q结点的左子树
            else q->rchild=s->lchild;//删除s结点只需将左子树置为q结点的右子树
            free(s);
    }
}
Status InitBSTree(BSTree *T){
    *T=NULL;
    return OK;
}
void InOderTraverse_LVR(BSTree T){
    if(T!=NULL){
        InOderTraverse_LVR(T->lchild);
        printf("%d  ",T->data);
        InOderTraverse_LVR(T->rchild);
    }
}
void main(){
    BSTree T1,sch;
    int i,s;
    if(InitBSTree(&T1)){
        printf("初始化成功！\n");
    }else{
        printf("初始化失败！\n");
    }
    RcdType data[N]={3,2,5,1,4};
    T1=NULL;
    for(i=0;i<N;i++){
        T1=InserBST(T1,data[i]);
    }
     printf("中序遍历\n");
     InOderTraverse_LVR(T1);
     printf("\n");

     if(SearchBST(T1,5)){
        printf("查询%d成功！\n");
     }else{
        printf("查询失败！\n");
    }

     printf("请输入要删除的元素：");
     scanf("%d",&s);
     sch=SearchBST(T1,s);
     if(DeleteBST(&sch,s)){
        printf("删除%d成功！\n");
     }else{
        printf("删除失败！\n");
    }
      printf("删除后的表的中序遍历\n");
     InOderTraverse_LVR(T1);

}