.7 二叉查找树的 建立 insert search remove 操作

//二叉查找树的建立，插入，删除操作
#include<stdio.h>
#include<stdlib.h>

typedef struct bsTree
{
    int  k ;
    struct bsTree * lchild , * rchild ;
}bstnode , * bstree ;

void create_tree(bstree *T,int k1)
{
     (*T) = (bstree)malloc(sizeof(bstnode));
     if(!(*T)){
        fprintf(stderr , "malloc error!
");
        exit(EXIT_FAILURE) ;
     }
     (*T)->k = k1 ;
     (*T)->lchild = (*T)->rchild = NULL;
}

bstree search(bstree T , int k )  //递归搜寻某个关键字为k 的节点
{
    if(!T) return NULL;
    if(k == T->k) return T ;
    if(k < T->k) return search(T->lchild , k ) ;
    if(k > T->k) return search(T->rchild , k ) ;
}

bstnode *itersearch(bstree T , int k ) //非递归搜寻某个关键字为k 的节点
{
    while(T){
        if(k == T->k) return T ;
        if(k < T->k) T = T->lchild  ;
        else T = T->rchild ;
    }
    return NULL;
}

bstree modifiedSearch(bstree T , int k1 ) //改进的搜寻 返回要insert的位置的上一个节点,用于插入操作
{
    bstree pre  = T ;
    while(T){
        if(k1 == T->k) return T ;
        pre = T ;
        if(k1 < T->k) T = T->lchild  ;
        else T = T->rchild ;
    }
    return pre;
}

bstree modifiedSearch2(bstree T , int k1 ) //改进的搜寻 返回要insert的位置的上一个节点 用于删除操作
{
    bstree pre  = T ;
    while(T){
        if(k1 == T->k) return pre ;
        pre = T ;
        if(k1 < T->k) T = T->lchild  ;
        else T = T->rchild ;
    }
    return NULL;
}

void insert(bstree T , bstree * tmp ,int k1 ) //插入操作  要用到modifiedSearch()函数
{
    bstree ptr ;
      (*tmp) = modifiedSearch(T,k1);
    if( *tmp ){
        ptr = (bstree)malloc(sizeof(bstnode)) ;
        ptr->k = k1 ;
        ptr->lchild = ptr->rchild  = NULL ;
            if(k1 < (*tmp)->k) (*tmp)->lchild = ptr ;
            else (*tmp)->rchild = ptr ;
            }
}

void in_order_traverse(bstree T) // 中序遍历 查找树
{
    if(T){

        in_order_traverse(T->lchild) ;
        printf("%d ",T->k) ;
        in_order_traverse(T->rchild) ;
    }
}

bstree remove1( bstree T , int k1)
{
    int f ;bstree Rf  = T,R  = NULL;
     Rf = modifiedSearch2(T,k1) ;
     if(k1 > Rf->k ) {
        f = 1 ; R = Rf->rchild ;
     }else{
        f = 0 ; R = Rf->lchild ;
     }

    if(!R->lchild && !R->rchild){
        if(f) Rf->rchild = NULL ;else Rf->lchild = NULL ;
    }
    else if(R->lchild || R->rchild ){
                if(f) Rf->rchild = NULL ;else Rf->lchild = NULL ;
    }
    else {
        bstree tmp  = R;
        while( tmp->lchild)
            tmp = R->lchild ;
    R->k = tmp->k ;
    R->lchild = NULL ;
    }
    return R ;
}

int main()
{
    bstree T ;
    int i ;
    create_tree(&T , 30) ;
        bstree tmp ;
       tmp = (bstree)malloc(sizeof(bstnode)) ;
       insert(T , &tmp ,5) ;
    bstree tmp1 ;
       tmp1 = (bstree)malloc(sizeof(bstnode)) ;
       insert(T , &tmp1 ,40) ;
       bstree tmp2 ;
       tmp2 = (bstree)malloc(sizeof(bstnode)) ;
       insert(T , &tmp2 ,35) ;
       bstree tmp3 ;
       tmp3 = (bstree)malloc(sizeof(bstnode)) ;
       insert(T , &tmp2 ,45) ;
       remove1(T,40);
    in_order_traverse(T);
    return 0 ;
}