初学数据结构（二叉树的基本运算）

初学数据结构（二叉树的基本运算）

《数据结构教程》第6版P243 7-1题

主函数

//
// Created by Snow on 2023/3/30.
//
#include"btree.cpp"
int main()
{
    BTNode *b;
    ElemType e;
    CreatBTree(b,"A(B(D,E(H(J,K(L,M(,N))))),C(F,G(,I)))");
    printf("输出二叉树b:\n");
    DispBTree(b);
    printf("\n");
    printf("输出'H'结点的左、右孩子结点值:\n");
    Print(LchildNode(FindBTree(b,'H')),e);
    printf("'H'结点的左孩子值为:%c\n",e);
    Print(RchildNode(FindBTree(b,'H')),e);
    printf("'H'结点的右孩子值为:%c\n",e);
    printf("二叉树的高度为:%d\n", BTHeight(b));
    printf("释放二叉树\n");
    DestroyBTree(b);
}

实现函数

//
// Created by Snow on 2023/3/30.
//
#include<cstdio>
#include<cstdlib>
#define MaxSize 50
typedef char ElemType;
typedef struct node
{
    ElemType data;
    struct node *lchild;
    struct node *rchild;
}BTNode;
void CreatBTree(BTNode *&b,char const *str)
{
    BTNode *St[MaxSize],*p;
    int top=-1,j=0,k;
    b=nullptr;
    char ch=str[j];
    while(ch!='\0')
    {
        switch(ch)
        {
            case '(':top++;St[top]=p;k=1;break;
            case ')':top--;break;
            case ',':k=2;break;
            default:
                p=(BTNode*)malloc(sizeof(BTNode));
                p->data=ch;
                p->lchild=p->rchild=nullptr;
                if(b==nullptr)
                    b=p;
                else
                {
                    switch(k)
                    {
                        case 1:St[top]->lchild=p;break;
                        case 2:St[top]->rchild=p;break;
                    }
                }
        }
        j++;
        ch=str[j];
    }
}
void DestroyBTree(BTNode *&b)
{
    if(b!=nullptr)
    {
        DestroyBTree(b->lchild);
        DestroyBTree(b->rchild);
        free(b);
    }
}
BTNode *FindBTree(BTNode *b,ElemType x)
{
    BTNode *p;
    if(b==nullptr)
        return nullptr;
    else if(b->data==x)
        return b;
    else
    {
        p=FindBTree(b->lchild,x);
        if(p!=nullptr)
            return p;
        else
            return FindBTree(b->rchild,x);
    }
}
BTNode *LchildNode(BTNode *p)
{
    return p->lchild;
}
BTNode *RchildNode(BTNode *p)
{
    return p->rchild;
}
int BTHeight(BTNode *b)
{
    int lchildh,rchildh;
    if(b==nullptr)
        return 0;
    else
    {
        lchildh=BTHeight(b->lchild);
        rchildh=BTHeight(b->rchild);
        return((lchildh>rchildh)?(lchildh+1):(rchildh+1));
    }
}
void DispBTree(BTNode *b)
{
    if(b!=nullptr)
    {
        printf("%c",b->data);
        if(b->lchild!=nullptr||b->rchild!=nullptr)
        {
            printf("(");
            DispBTree(b->lchild);
            if(b->rchild!=nullptr)
                printf(",");
            DispBTree(b->rchild);
            printf(")");
        }
    }
}
//没什么阮用，主要是上面找孩子的函数定义问题，也可以不要这个，直接去找孩子那个函数里输出
void Print(BTNode *p,ElemType &e)
{
    e=p->data;
}