《数据结构》二叉树递归遍历

 

 

/*******************************************************************************
/* <PRE>
/* 版权所有    : -
/* 模块名      : 树
/* 文件名      : btree.cpp
/* 功能描述    : 二叉树的递归遍历
/* 作者        : <xxx>
/* 版本        : 1.0
/* -----------------------------------------------------------------------------
/* 备注        : 输入示例与输出结果
/* 
/* e.g. input  : ABD###CE#F###
/* bi-tree     :
/*                 A
/*                / \
/*               B   C
/*              /   /
/*             D   E
/*                  \
/*                   F
/* 
/* pre-order traverse: A B D C E F
/* in-order traverse: D B A E F C
/* post-order traverse: D B F E C A
/* -----------------------------------------------------------------------------
/* 修改记录    :
/* 日 期        版本     修改人        修改内容
/* 2011/01/01   1.0      <xxx>         创建
/* </PRE>
*******************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string>

/******************************************************************************
/* 数据类型和常量定义
/******************************************************************************/
#define OK           1
#define ERROR        0
#define OVERFLOW    -2

typedef int Status;
typedef int TElemType;


/******************************************************************************
/* 数据结构声明
/******************************************************************************/
/* 二叉树的链式存储结构 */
typedef struct BiTNode {
    TElemType data;
    struct BiTNode *lchild, *rchild; /* 左右孩子指针 */
} BiTNode, *BiTree;


/******************************************************************************
/* 函数原型声明
/******************************************************************************/
Status Visit(TElemType e);
Status PreOrderTraverse(BiTree T, Status (*Visit)(TElemType));
Status InOrderTraverse(BiTree T, Status (*Visit)(TElemType));
Status PostOrderTraverse(BiTree T, Status (*Visit)(TElemType));


/*******************************************************************************
/* <FUNC>
/* 函数名   : Visit
/* 功能     : 打印节点数据
/* 参数     : -
/* 返回值   : -
/* 备注     : -
/* 作者     : <xxx>
/* </FUNC>
*******************************************************************************/
Status Visit(TElemType e)
{
    printf("%c", e);
    return OK;
}

/*******************************************************************************
/* <FUNC>
/* 函数名   : PreOrderTraverse
/* 功能     : 前序遍历二叉树
/* 参数     : -
/* 返回值   : -
/* 备注     : 递归法
/* 作者     : <xxx>
/* </FUNC>
*******************************************************************************/
Status PreOrderTraverse(BiTree T, Status (*Visit)(TElemType))
{
    if(T){
        if(Visit(T->data))
            if(PreOrderTraverse(T->lchild, Visit))
                if(PreOrderTraverse(T->rchild, Visit))
                    return OK;
                return ERROR;
    }
    else
        return OK;
}

/*******************************************************************************
/* <FUNC>
/* 函数名   : InOrderTraverse
/* 功能     : 中序遍历二叉树
/* 参数     : -
/* 返回值   : -
/* 备注     : 递归法
/* 作者     : <xxx>
/* </FUNC>
*******************************************************************************/
Status InOrderTraverse(BiTree T, Status (*Visit)(TElemType))
{
    if(T){
        if(InOrderTraverse(T->lchild, Visit))
            if(Visit(T->data))
                if(InOrderTraverse(T->rchild, Visit))
                    return OK;
                return ERROR;
    }
    else
        return OK;
}

/*******************************************************************************
/* <FUNC>
/* 函数名   : PostOrderTraverse
/* 功能     : 后序遍历二叉树
/* 参数     : -
/* 返回值   : -
/* 备注     : 递归法
/* 作者     : <xxx>
/* </FUNC>
*******************************************************************************/
Status PostOrderTraverse(BiTree T, Status (*Visit)(TElemType))
{
    if(T){
        if(PostOrderTraverse(T->lchild, Visit))
            if(PostOrderTraverse(T->rchild, Visit))
                if(Visit(T->data))
                    return OK;
                return ERROR;
    }
    else
        return OK;
}

/*******************************************************************************
/* <FUNC>
/* 函数名   : CreateBiTree
/* 功能     : 创建二叉树
/* 参数     : -
/* 返回值   : -
/* 备注     : 前序方式创建
/* 作者     : <xxx>
/* </FUNC>
*******************************************************************************/
Status CreateBiTree(BiTree &T)
{
    char ch = getchar();
    if('#' == ch) T = NULL;
    else {
        if(!(T = (BiTNode *)malloc(sizeof(BiTNode)))) exit(OVERFLOW);
        T->data = ch;             //生成根结点
        CreateBiTree(T->lchild);  //构造左子树
        CreateBiTree(T->rchild);  //构造右子树
    }
    return OK;
}

/*******************************************************************************
/* <FUNC>
/* 函数名   : main
/* 功能     : 测试函数
/* 参数     : -
/* 返回值   : -
/* 备注     : -
/* 作者     : <xxx>
/* </FUNC>
*******************************************************************************/
void main()
{
    BiTree T;
    CreateBiTree(T);

    printf("Pre Order Traverse: ");
    PreOrderTraverse(T, Visit);        //前序方式遍历二叉树

    printf("\nIn Order Traverse: ");
    InOrderTraverse(T, Visit);         //中序方式遍历二叉树

    printf("\nPost Order Traverse: ");
    PostOrderTraverse(T, Visit);       //后序方式遍历二叉树
}