数据结构之二叉树（C语言实现）

/*
********************************************************************************
*                                数据结构
*                               
*
*
*                                作者: zhi-z
* 
*                                QQ号: 2282643301
* 
********************************************************************************
*文件名     : BinTree.c 
*作用       : 操作二叉树 
*原理       :
********************************************************************************
*版本     作者            日期            说明
*V0.1            		2016-10-2      初始版本
********************************************************************************
*/


#include <stdio.h> 
#include <stdlib.h> 
#define QUEUE_MAXSIZE 50

/********************************************************************************
*数据结构名 : ChainTree
*参数		：
*作用       : 定义二叉树的结构 
*内容		：
*
********************************************************************************/
typedef char DATA;       //定义元素类型 
typedef struct ChainTree  //定义二叉树结点类型 
{
    DATA data;	//元素数据
    struct ChainTree *left;	//左子树结点指针
    struct ChainTree *right;	//右子树结点指针
}ChainBinTree;

/********************************************************************************
*函数名     : BinTreeInit
*参数		：无 
*作用       : 初始化二叉树 
*内容		：
*
********************************************************************************/
ChainBinTree *BinTreeInit(ChainBinTree *node) //初始化二叉树根结点 
{
     if(node!=NULL) //若二叉树根结点不为空 
         return node;
     else
         return NULL;
}

/********************************************************************************
*函数名     : BinTreeInit
*参数		：无 
*作用       : 添加数据到二叉树 
*内容		：
*
********************************************************************************/
int BinTreeAddNode(ChainBinTree *bt,ChainBinTree *node,int n) //添加数据到二叉树 
//bt为父结点，node为子结点,n=1表示添加左子树，n=2表示添加右子树 
{
    if(bt==NULL)
    {
        printf("父结点不存在，请先设置父结点!\n");
        return 0;
    } 
    switch(n)
    {
        case 1: //添加到左结点 
            if(bt->left) //左子树不为空 
            {
                printf("左子树结点不为空!\n");
                return 0;
            }else
                bt->left=node;
            break;
        case 2://添加到右结点
            if( bt->right) //右子树不为空 
            {
                printf("右子树结点不为空!\n"); 
                return 0;
            }else
                bt->right=node;
            break;
        default:
            printf("参数错误!\n");
            return 0;
    }
    return 1;
}

/********************************************************************************
*函数名     : BinTreeInit
*参数		：无 
*作用       : 返回左子结点 
*内容		：
*
********************************************************************************/
ChainBinTree *BinTreeLeft(ChainBinTree *bt) //返回左子结点 
{
    if(bt)
        return bt->left;
    else
        return NULL;
}

/********************************************************************************
*函数名     : BinTreeInit
*参数		：无 
*作用       : 返回右子结点 
*内容		：
*
********************************************************************************/
ChainBinTree *BinTreeRight(ChainBinTree *bt) //返回右子结点 
{
    if(bt)
        return bt->right;
    else
        return NULL;
}

/********************************************************************************
*函数名     : BinTreeInit
*参数		：无 
*作用       : 检查二叉树是否为空，为空则返回1,否则返回0 
*内容		：
*
********************************************************************************/
int BinTreeIsEmpty(ChainBinTree *bt) //检查二叉树是否为空，为空则返回1,否则返回0 
{
    if(bt)
        return 0;
    else
        return 1;
} 

/********************************************************************************
*函数名     : BinTreeInit
*参数		：无 
*作用       : 求二叉树深度
*内容		：
*
********************************************************************************/
int BinTreeDepth(ChainBinTree *bt) //求二叉树深度
{
    int dep1,dep2;
    if(bt==NULL)
        return 0; //对于空树，深度为0
    else
    {
        dep1 = BinTreeDepth(bt->left); //左子树深度 (递归调用)
        dep2 = BinTreeDepth(bt->right); //右子树深度 (递归调用)
        if(dep1>dep2)
           return dep1 + 1; 
        else
            return dep2 + 1; 
    } 
} 

/********************************************************************************
*函数名     : BinTreeInit
*参数		：无 
*作用       : 在二叉树中查找值为data的结点 
*内容		：
*
********************************************************************************/
ChainBinTree *BinTreeFind(ChainBinTree *bt,DATA data) //在二叉树中查找值为data的结点 
{
    ChainBinTree *p;
    if(bt==NULL)
        return NULL;
    else
    {
        if(bt->data==data)
            return bt; 
        else{ // 分别向左右子树递归查找 
            if(p=BinTreeFind(bt->left,data))
                return p;
            else if(p=BinTreeFind(bt->right, data)) 
                return p;
            else
                return NULL; 
        } 
    } 
} 

/********************************************************************************
*函数名     : BinTreeClear
*参数		：无 
*作用       : 清空二叉树，使之变为一棵空树
*内容		：
*
********************************************************************************/
void BinTreeClear(ChainBinTree *bt) // 清空二叉树，使之变为一棵空树
{
     if(bt)
     {
         BinTreeClear(bt->left); //清空左子树 
         BinTreeClear(bt->right);//清空右子树 
         free(bt);//释放当前结点所占内存 
         bt=NULL;
     }
     return; 
}

/********************************************************************************
*函数名     : BinTreeInit
*参数		：
*作用       : 二叉树的遍历（先序遍历二叉树） 
*内容		：
*
********************************************************************************/
void BinTree_DLR(ChainBinTree *bt,void (*oper)(ChainBinTree *p))  //先序遍历 
{     
     if(bt)//树不为空，则执行如下操作 
     {
         oper(bt); //处理结点的数据 
         BinTree_DLR(bt->left,oper);
         BinTree_DLR(bt->right,oper);
     }
     return; 
} 

/********************************************************************************
*函数名     : BinTree_LDR
*参数		：
*作用       : 二叉树的遍历（中序遍历二叉树） 
*内容		：
*
********************************************************************************/
void BinTree_LDR(ChainBinTree *bt,void(*oper)(ChainBinTree *p))  //中序遍历 
{
     if(bt)//树不为空，则执行如下操作 
     {
         BinTree_LDR(bt->left,oper); //中序遍历左子树
         oper(bt);//处理结点数据 
         BinTree_LDR(bt->right,oper); //中序遍历右子树/
     }
     return; 
} 

/********************************************************************************
*函数名     : BinTree_LRD
*参数		：
*作用       : 二叉树的遍历（后序遍历二叉树） 
*内容		：
*
********************************************************************************/
void BinTree_LRD(ChainBinTree *bt,void (*oper)(ChainBinTree *p)) //后序遍历
{
     if(bt)
     {
         BinTree_LRD(bt->left,oper); //后序遍历左子树 
         BinTree_LRD(bt->right,oper); //后序遍历右子树/
         oper(bt); //处理结点数据
     }
     return; 
}


/********************************************************************************
*函数名     : oper
*参数		：
*作用       : 操作二叉树结点数据 
*内容		：
*
********************************************************************************/
void oper(ChainBinTree *p) //操作二叉树结点数据 
{
     printf("%c ",p->data); //输出数据
     return;
}


/********************************************************************************
*函数名     : BinTree_Level
*参数		：
*作用       : 按层遍历 
*内容		：
*
********************************************************************************/
void BinTree_Level(ChainBinTree *bt,void (*oper)(ChainBinTree *p)) //按层遍历 
{
     ChainBinTree *p;
     ChainBinTree *q[QUEUE_MAXSIZE]; //定义一个顺序栈 
     int head=0,tail=0;//队首、队尾序号 
     if(bt)//若队首指针不为空     
     {
         tail=(tail+1)%QUEUE_MAXSIZE;//计算循环队列队尾序号 
         q[tail] = bt;//将二叉树根指针进队
     }
     while(head!=tail) //队列不为空，进行循环 
     {
         head=(head+1)%QUEUE_MAXSIZE; //计算循环队列的队首序号 
         p=q[head]; //获取队首元素 
         oper(p);//处理队首元素 
         if(p->left!=NULL) //若结点存在左子树，则左子树指针进队 
         {
             tail=(tail+1)%QUEUE_MAXSIZE;//计算循环队列的队尾序号 
             q[tail]=p->left;//将左子树指针进队 
         }
                 
         if(p->right!=NULL)//若结点存在右孩子，则右孩子结点指针进队 
         {
             tail=(tail+1)%QUEUE_MAXSIZE;//计算循环队列的队尾序号 
             q[tail]=p->right;//将右子树指针进队 
         }
     }
     return; 
}

/*
********************************************************************************
*                                数据结构
*                               
*
*
*                                作者:zhi-z
* 
*                                QQ号:2282643301
* 
********************************************************************************
*文件名     : BinTreeTest.c 
*作用       : 二叉树操作实现测试程序 
*原理       :
********************************************************************************
*版本     作者            日期            说明
*V0.1            		2016-10-3     初始版本
********************************************************************************
*/

#include <stdio.h> 
#include "BinTree.c"


void oper1(ChainBinTree *p) //操作二叉树结点数据 
{
     printf("%c ",p->data); //输出数据
     return;
}
ChainBinTree *InitRoot()  //初始化二叉树的根 
{
    ChainBinTree *node;
    if(node=(ChainBinTree *)malloc(sizeof(ChainBinTree))) //分配内存
    {
        printf("\n输入根结点数据:");
        scanf("%s",&node->data);
        node->left=NULL;
        node->right=NULL;
        return BinTreeInit(node);
    }
    return NULL;
}
void AddNode(ChainBinTree *bt)
{
     ChainBinTree *node,*parent;
     DATA data;
     char select;
    if(node=(ChainBinTree *)malloc(sizeof(ChainBinTree))) //分配内存
    {
        printf("\n输入二叉树结点数据:");
        fflush(stdin);//清空输入缓冲区 
        scanf("%s",&node->data);
        node->left=NULL; //设置左右子树为空 
        node->right=NULL;
        

        printf("输入父结点数据:");
        fflush(stdin);//清空输入缓冲区 
        scanf("%s",&data);
        parent=BinTreeFind(bt,data);//查找指定数据的结点 
        if(!parent)//若未找到指定数据的结点 
        {
            printf("未找到父结点!\n");
            free(node); //释放创建的结点内存 
            return;
         }
         printf("1.添加到左子树\n2.添加到右子树\n");
         do{
            select=getch();
            select-='0';
            if(select==1 || select==2)
                BinTreeAddNode(parent,node,select); //添加结点到二叉树 
         }while(select!=1 && select!=2);
    }
    return ;
}
int main() 
{
    ChainBinTree *root=NULL; //root为指向二叉树根结点的指针 
    char select;
    void (*oper1)(); //指向函数的指针 
    oper1=oper; //指向具体操作的函数 
    do{
        printf("\n1.设置二叉树根元素    2.添加二叉树结点\n");
        printf("3.先序遍历            4.中序遍历\n");
        printf("5.后序遍历            6.按层遍历\n");
        printf("7.二叉树深度          0.退出\n");
        select=getch();
        switch(select){
        case '1': //设置根元素 
             root=InitRoot();
             break;
        case '2': //添加结点 
             AddNode(root);
             break;
        case '3'://先序遍历
             printf("\n先序遍历的结果：");
             BinTree_DLR(root,oper1);
             printf("\n");
             break;
        case '4'://中序遍历 
             printf("\n中序遍历的结果：");
             BinTree_LDR(root,oper1);
             printf("\n");
             break;
        case '5'://后序遍历 
             printf("\n后序遍历的结果：");
             BinTree_LRD(root,oper1);
             printf("\n");
             break;
        case '6'://按层遍历
             printf("\n按层遍历的结果：");
             BinTree_Level(root,oper1);
             printf("\n");
             break;
        case '7'://二叉树的深度
            printf("\n二叉树深度为:%d\n",BinTreeDepth(root));
            break; 
        case '0':
             break;
        }
    }while(select!='0');
    BinTreeClear(root);//清空二叉树 
    root=NULL;
    getch();
    return 0;
}