PTA 6-09 二叉树的非递归遍历

题目

本题要求用非递归的方法实现对给定二叉树的 3 种遍历。

函数接口定义：

void InorderTraversal( BinTree BT );
void PreorderTraversal( BinTree BT );
void PostorderTraversal( BinTree BT );

其中BinTree结构定义如下：

typedef struct TNode *Position;
typedef Position BinTree;
struct TNode{
    ElementType Data;
    BinTree Left;
    BinTree Right;
    int flag;
};

要求 3 个函数分别按照访问顺序打印出结点的内容，格式为一个空格跟着一个字符。

此外，裁判程序中给出了堆栈的全套操作，可以直接调用。

裁判测试程序样例：

#include <stdio.h>
#include <stdlib.h>
typedef enum { false, true } bool;

typedef char ElementType;
typedef struct TNode *Position;
typedef Position BinTree;
struct TNode{
    ElementType Data;
    BinTree Left;
    BinTree Right;
    int flag;
};

/*------堆栈的定义-------*/
typedef Position SElementType;
typedef struct SNode *PtrToSNode;
struct SNode {
    SElementType Data;
    PtrToSNode Next;
};
typedef PtrToSNode Stack;

/* 裁判实现，细节不表 */
Stack CreateStack();
bool IsEmpty( Stack S );
bool Push( Stack S, SElementType X );
SElementType Pop( Stack S ); /* 删除并仅返回S的栈顶元素 */
SElementType Peek( Stack S );/* 仅返回S的栈顶元素 */
/*----堆栈的定义结束-----*/

BinTree CreateBinTree(); /* 裁判实现，细节不表 */
void InorderTraversal( BinTree BT );
void PreorderTraversal( BinTree BT );
void PostorderTraversal( BinTree BT );

int main()
{
    BinTree BT = CreateBinTree();
    printf("Inorder:");    InorderTraversal(BT);    printf("\n");
    printf("Preorder:");   PreorderTraversal(BT);   printf("\n");
    printf("Postorder:");  PostorderTraversal(BT);  printf("\n");
    return 0;
}
/* 你的代码将被嵌在这里 */

输入样例：

输出样例：

Inorder: D B E F A G H C I
Preorder: A B D F E C G H I
Postorder: D E F B H G I C A

一个思路

 每个元素的输出，在局部里都是根，如B是D,F的根，D是左NULL和右NULL的根。

以中序为例：

先向左一直遍历并入栈(栈底到栈顶元素为ABD)，直到NULL，此时p变栈顶D并出栈输出，即输出D并删除,然后p变右孩，即D的右NULL。再次进入循环，p为NULL，所以p变栈顶并出栈输出，即B,然后p变右孩，即F。F不空，入栈，变左孩E，以后过程以此类推。

其中输出D的时候，D是D的左NULL和右NULL的根，输出B的时候，B是B的左孩D和右孩F的根。

对于后序：

注意的点是标明回到局部根时是从左回还是右回。这里把向左遍历设为1，向右遍历设为-1。

结果

分别为中序，先序，后序。

void InorderTraversal( BinTree BT )
{
    Stack S;
    BinTree p=BT;
    if(BT== NULL)
        return ;
    S=CreateStack();
    while(p||!IsEmpty(S))//结束条件为p为NULL且栈空
    {
        //若p不空，入栈，p变左孩子
        if(p)
        {
            Push(S,p);
            p=p->Left;
        }
        //若p空，变为栈顶元素，出栈，p变右孩子
        else
        {
            p=Pop(S);
            printf(" %c",p->Data);
            p=p->Right;
        }
    }
}
void PreorderTraversal( BinTree BT )
{
    Stack S;
    if(BT==NULL)
        return;
    BinTree p=BT;
    S=CreateStack();
    while(p||!IsEmpty(S))//结束条件为p为NULL且栈空
    {
        //若p不空，先输出p(此时p为根)，p入栈(便于等会通过p找到p的右孩子)，p变左孩子
        if(p)
        {
            printf(" %c",p->Data);
            Push(S,p);
            p=p->Left;
        }
        //若p空，变为栈顶元素，出栈，p变右孩子
        else
        {
            p=Pop(S);
            p=p->Right;
        }
    }
}
void PostorderTraversal( BinTree BT )
{
    Stack S;
    if(BT==NULL)
        return;
    BinTree p=BT;
    S=CreateStack();
    while(p||!IsEmpty(S))//结束条件为p为NULL且栈空
    {
        //若p不空，p->flag设1（表明向左遍历），p入栈，p变左孩子，向左遍历
        if(p)
        {
            p->flag=1;
            Push(S,p);
            p=p->Left;
        }
        //若p空，p变栈顶元素，若flag=1，表明从左回到局部根，把flag设为-1，p变右孩子，向右遍历
        //若flag=-1，表明从右回到局部根，输出局部根，p设为NULL（为了在下次循环时把p变为栈顶元素）
        else
        {
            p=Peek(S);
            if(p->flag==1)
            {
                p->flag=-1;
                p=p->Right;
            }
            else if(p->flag==-1)
            {
                printf(" %c",p->Data);
                p=Pop(S);
                p=NULL;
            }
        }
    }
}