二叉树的基本操作和实现

这段代码是一个使用 C 语言实现的二叉树操作程序，具体的功能包括建立二叉树、先序遍历、中序遍历、后序遍历、计算二叉树深度、判断二叉树是否为满二叉树等。用户可以根据操作菜单选择相应的功能进行操作。代码中包含了处理用户输入错误的情况，以及在程序结束前销毁二叉树等功能。

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

typedef char TElemType;
typedef int Status;

typedef struct BiTNode {
    TElemType data;
    struct BiTNode* lchild, * rchild;
} BiTNode, * BiTree;

// 建立二叉树
Status CreateBiTree(BiTree& T)
{
    char ch;
    scanf("%c", &ch);
    if (ch == '#')
    {
        T = NULL;
    }
    else
    {
        if (!(T = (BiTNode*)malloc(sizeof(BiTNode))))
        {
            exit(1);
        }
        T->data = ch;
        CreateBiTree(T->lchild);
        CreateBiTree(T->rchild);
    }

    return 1;
}

// 后序遍历
Status LRD(BiTree T)
{
    if (T == NULL)
    {
        return 0;
    }
    else
    {
        LRD(T->lchild);
        LRD(T->rchild);
        printf("%c ", T->data);
    }

    return 1;
}

// 中序遍历
Status LDR(BiTree T)
{
    if (T == NULL)
    {
        return 0;
    }
    else
    {
        LDR(T->lchild);
        printf("%c ", T->data);
        LDR(T->rchild);
    }

    return 1;
}

// 先序遍历
Status DLR(BiTree T)
{
    if (T == NULL)
    {
        return 0;
    }
    else
    {
        printf("%c ", T->data);
        DLR(T->lchild);
        DLR(T->rchild);
    }

    return 1;
}

// 销毁二叉树
void DestroyBiTree(BiTree& T)
{
    if (T == NULL)
    {
        return;
    }
    DestroyBiTree(T->lchild);
    DestroyBiTree(T->rchild);
    free(T);
    T = NULL;
}

// 计算二叉树深度
int GetTreeDepth(BiTree T)
{
    if (T == NULL)
    {
        return 0;
    }
    else
    {
        int leftDepth = GetTreeDepth(T->lchild);
        int rightDepth = GetTreeDepth(T->rchild);
        return (leftDepth > rightDepth) ? (leftDepth + 1) : (rightDepth + 1);
    }
}

// 判断二叉树是否为满二叉树
int IsFullBinaryTree(BiTree T)
{
    if (T == NULL)
    {
        return 1;
    }

    int leftHeight = GetTreeDepth(T->lchild);
    int rightHeight = GetTreeDepth(T->rchild);

    if (leftHeight != rightHeight)
    {
        return 0;
    }

    return IsFullBinaryTree(T->lchild) && IsFullBinaryTree(T->rchild);
}

// 操作菜单
void OperateMenu()
{
    printf("-------操作菜单--------\n");
    printf("0. 退出程序\n");
    printf("1. DLR遍历\n");
    printf("2. LDR遍历\n");
    printf("3. LRD遍历\n");
    printf("4. 计算二叉树深度\n");
    printf("5. 销毁二叉树\n");
    printf("6. 判断二叉树是否为满二叉树\n");
    printf("----------------------\n");
    printf("请输入操作代码：\n");
}

int main()
{
    BiTree T = NULL;
    CreateBiTree(T);

    int choice;
    do {
        OperateMenu();
        if (scanf("%d", &choice) != 1) {
            printf("输入错误，请输入一个有效的操作代码！\n");
            scanf("%*s");  // 清除输入缓冲区中的无效字符
            continue;
        }

        switch (choice) {
        case 0:
            printf("退出程序\n");
            break;
        case 1:
            printf("DLR遍历结果：");
            DLR(T); // 先序遍历二叉树
            printf("\n");
            break;
        case 2:
            printf("LDR遍历结果：");
            LDR(T); // 中序遍历二叉树
            printf("\n");
            break;
        case 3:
            printf("LRD遍历结果：");
            LRD(T); // 后序遍历二叉树
            printf("\n");
            break;
        case 4:
            printf("二叉树深度为：%d\n", GetTreeDepth(T)); // 计算二叉树深度
            break;
        case 5:
            printf("销毁二叉树\n");
            DestroyBiTree(T); // 销毁二叉树
            break;
        case 6:
            if (!IsFullBinaryTree(T))
            {
                printf("该二叉树是满二叉树\n");
            }
            else
            {
                printf("该二叉树不是满二叉树\n");
            }
            break;
        default:
            printf("输入错误，请输入一个有效的操作代码！\n");
            break;
        }
    } while (choice != 0);

    DestroyBiTree(T); // 在程序结束前销毁二叉树

    return 0;
}

这段代码实现了一个二叉树操作程序。代码包括以下几个部分：

1. 定义了一个二叉树节点的结构体 BiTNode，包含数据项和指向左子树和右子树的指针。
2. 定义了一个二叉树的别名 BiTree，为 BiTNode 的指针类型。
3. 创建了一个函数 CreateBiTree，用于建立二叉树。它通过递归方式输入节点的数据，并创建左子树和右子树。
4. 实现了三种遍历二叉树的函数：DLR、LDR 和 LRD。
   • DLR 函数实现了先序遍历，先输出根节点的数据，然后遍历左子树，最后遍历右子树。
   • LDR 函数实现了中序遍历，先遍历左子树，然后输出根节点的数据，最后遍历右子树。
   • LRD 函数实现了后序遍历，先遍历左子树，然后遍历右子树，最后输出根节点的数据。
5. 实现了计算二叉树深度的函数 GetTreeDepth，使用递归方式计算左子树和右子树的深度，并取较大值加1作为整个树的深度。
6. 实现了判断二叉树是否为满二叉树的函数 IsFullBinaryTree。判断条件是左子树和右子树的深度相等，且左子树和右子树都是满二叉树。
7. 实现了一个操作菜单函数 OperateMenu，用于显示可选操作。
8. 在 main 函数中，首先调用 CreateBiTree 函数创建二叉树，然后进入一个循环，根据用户选择的操作代码执行相应的操作，直到用户选择退出程序。
9. 在程序结束前，调用 DestroyBiTree 函数销毁二叉树。

这段代码可以实现对二叉树进行各种操作，如遍历、计算深度和判断是否为满二叉树等。用户可以根据操作菜单选择所需的操作。