二叉排序树_创建插入删除

二叉排序树的学习

经验教训：

①结构体指针声明，最前面要加struct，例如struct BSTNode *lchild;，而不是BSTNode *lchild;。

②typedef的作用：用来为复杂的声明定义简单的别名。例如，typedef struct BSTNode BST;中BST的作用等效于struct BSTNode。

③声明变量其为指针时，命名前用p，表面其是指针，会更好更直观。

④删除结点后，需要考虑到双亲结点的指向 或者 置空NULL。

 

#include <stdio.h>
#include <stdlib.h>

#define TRUE 1  
#define FALSE 0  

typedef int elemtype;

struct BSTNode
{
    elemtype data;
    struct BSTNode *lchild;
    struct BSTNode *rchild;
};

typedef struct BSTNode BST;
typedef struct BSTNode * pBST;

pBST CreatBST()
{
    pBST T;
    elemtype data;
    printf("input value of root data:");
    scanf("%d",&data);
    if(!(T = (pBST)malloc(sizeof(BST)) ))
    {
        printf("overflow!\n");
        return NULL;
    }
    else
    {
        T->data = data;
        T->lchild = NULL;
        T->rchild = NULL;
    }
    return T;
}

pBST InsertBST(pBST q, elemtype insertData)
{
    if(q == NULL)
    {
        if(!(q = (pBST)malloc(sizeof(struct BSTNode)) ))
        {
            printf("overflow!\n");
            return NULL;
        }
        q->data = insertData;
        q->lchild = NULL;
        q->rchild = NULL;
    }
    else if( insertData < q->data)
    {
        q->lchild = InsertBST(q->lchild, insertData);
    }
    else
    {
        q->rchild = InsertBST(q->rchild, insertData);
    }
    return q;
}

void Inorder(pBST root)
{
    if(root)
    {
        Inorder(root->lchild);
        printf("%d ",root->data);
        Inorder(root->rchild);
    }
}

//查找X,若成功返回结点地址，失败返回NULL 
pBST Find(pBST root, elemtype findData)
{
    while(root) {
        if(findData < root->data) 
            root = root->lchild;
        else if(findData > root->data)
            root = root->rchild;
        else
            return root;
    }
    return NULL;
}
void DeletNode(pBST root, elemtype deletData)  
{
    pBST p = Find(root,deletData);
    if (p->lchild)      //有左孩子或者有两个孩子 
    {
        pBST r = p->lchild;   //r指向其左子树;  
        pBST prer = p->lchild;   //prer指向其左子树;  
        while(r->rchild != NULL)//搜索左子树的最右边的叶子结点r  
        {
            prer = r;  
            r = r->rchild;  
        }  
        p->data = r->data;  
  
        if(prer != r)//若r不是p的左孩子，把r的左孩子作为r的父亲的右孩子  
            prer->rchild = r->lchild;  
        else          // r是p的左孩子，p的左子树最大值就是r,且r没有右子树 
            p->lchild = r->lchild; //否则结点p的左子树指向r的左子树  
  
        free(r);  
    }  
    else if(p->rchild)    //只有右孩子 
    {  
        pBST r = p->rchild;   //q指向其右子树;  
        pBST prer = p->rchild;   //prer指向其右子树; 
        while(r->lchild != NULL) //搜索右子树的最左边的叶子结点r 
        {
            prer = r;  
            r = r->lchild;
        }
        p->data = r->data; 
        
        if(prer != r)//若r不是p的右孩子，把r的右孩子作为r的父亲的左孩子  
            prer->lchild = r->rchild;  
        else          // r是p的右孩子，p的右子树最小值就是r,且r没有左子树 
            p->rchild = r->rchild; //否则结点p的左子树指向r的左子树  
            
         free(r);  
    } 
    else    //叶子结点 
    {
        pBST parent;
        while(root)
        {
            if(p->data > root->data)
             {
                 parent = root;
                 root = root->rchild;
             }
             else if(p->data < root->data)
             {
                 parent = root;
                 root = root->lchild;
             }
             else
                 break;
        }
        if(p->data < parent->data)//如果p是双亲的左孩子
             parent->lchild = NULL;
        else
            parent->rchild = NULL;
        free(p);
    } 
}  


int main()
{
    pBST tree;
    int countNode = 0, i,deletData;
    elemtype insertData;

    pBST findNode;
    pBST parent;

    tree = CreatBST();
    printf("input number of node, you wanted:");
    scanf("%d",&countNode);
    for(i = 0; i < countNode; i++)
    {
        scanf("%d",&insertData);
        InsertBST(tree,insertData);
    }
    Inorder(tree);
    printf("\n");
    
    printf("delete number of node, you wanted:");
    scanf("%d",&deletData);
    DeletNode(tree,deletData);
    Inorder(tree);
    printf("\n");
    return 0;
}