前期准备:引入头文件、define、构建树
#include<stdio.h>
#include<stdlib.h>
#include<stdbool.h>
#define ElemType char
#define MAXSIZE 100//栈中
#define SElemType BiTree//非递归栈中要用
//构建结点
typedef struct BiTreeNode{
ElemType data;//结点内的数据
BiTreeNode *lchild;//左孩子结点
BiTreeNode *rchild;//右孩子结点
bool flag = false;//是否访问过该结点的rchild(后序要用)
}BiTNode,*BiTree;//指向BiNode类型的指针变量BiTree
//栈
typedef struct{
SElemType data[MAXSIZE];//静态数组存放栈中元素
int top;//栈顶指针
}SqStack;//Sq:sequence顺序
void InitStack(SqStack &S){
S.top = -1;//初始化栈顶指针
}
void PushStack(SqStack &S,SElemType x){
if(S.top == MAXSIZE - 1)//栈满报错
return;
S.data[++S.top] = x;//指针先+1
}
void PopStack(SqStack &S,SElemType &x){
if(S.top == -1)//栈空报错
return;
x = S.data[S.top--];//栈顶元素先出栈,指针-1
}
void GetTopStack(SqStack &S,SElemType &x){
if(S.top == -1)//栈空报错
return;
x = S.data[S.top];//栈顶元素先出栈,指针-1
}
//访问函数
void visit(BiTree T){
printf("%c",T->data);
}
//创建树
void CreateTree(BiTree &T){
printf("请输入结点:\n");
char ch = getchar();//读取输入的字符
getchar();//吃掉回车
if(ch == ' '){//输入空格代表NULL
T = NULL;
}
else{
T = (BiTree)malloc(sizeof(BiTNode));//T为BiTree类型,是指向BiTNode的指针,给他malloc一个BiTNode结构体
T->data = ch;
CreateTree(T->lchild);//遍历
CreateTree(T->rchild);
}
}
中序非递归:
以下图为例,先将根节点A入栈,令x变量初始为根节点A,开始循环。
栈S非空,开始第1遍循环,沿着lchild方向一直入栈,即将B入栈,将D入栈。此时x->lchild为空,即D的lchild为空。接着执行pop语句将栈顶元素出栈并把出栈的结点赋给x,打印x,也就是D。如果x的rchild存在,即D存在rchild,将D的rchild也就是G入栈,把结点G赋给变量x,结束第1遍循环。
栈S非空,开始第2遍循环,结点G没有lchild,跳过循环里的第一个while。接着执行pop语句将栈顶元素出栈并把出栈的结点赋给x,打印x,也就是G。G的rchild不存在,结束第2遍循环。
栈S非空,开始第3遍循环,结点G没有lchild,跳过循环里的第一个while。接着执行pop语句将栈顶元素出栈并把出栈的结点赋给x,打印x,也就是B。如果x的rchild存在,即B存在rchild,将B的rchild也就是E入栈,把结点E赋给变量x,结束第3遍循环。
栈S非空,开始第4遍循环,结点E没有lchild,跳过循环里的第一个while。接着执行pop语句将栈顶元素出栈并把出栈的结点赋给x,打印x,也就是E。E的rchild不存在,结束第4遍循环。
以此类推....
void In(BiTree T){
SqStack S;
SElemType x;
InitStack(S);//初始化栈
PushStack(S,T);//把根节点入栈
x = T;//x=根节点
while(S.top != -1){//当栈非空时
//沿着lchild方向一直入栈,直到没有lchild
while(x->lchild){//如果结点的lchild存在
x = x->lchild;//一直向左下走
PushStack(S,x);//入栈
}
PopStack(S,x);//出栈
visit(x);//打印
if(x->rchild) {//如果已经出栈的元素有rchild
x = x->rchild;
PushStack(S,x); //将rchild入栈
}//结束本次循环
}
}
后序非递归遍历
以下图为例,先将根节点A入栈,令x变量初始为根节点A,开始循环。
与中序遍历相比,后序遍历只多了一个flag的使用,flag初始置为false,即该结点没有访问过rchild。在循环中pop一个元素的时候,因为是后序遍历(左右根,即根节点最后才输出),所以如果没有访问过rchild,被pop之后还要把它重新push进去,然后把flag置成true代表已经访问过右子树了。读者可按照以下代码自己画出循环图理解。
void Post(BiTree T){
SqStack S;
SElemType x;
InitStack(S);//初始化栈
PushStack(S,T);//将根结点入栈
x = T;//把根节点赋值给变量x
while(S.top != -1){
while(x && x->lchild){
x = x->lchild;
PushStack(S,x);
}
PopStack(S,x);//出栈
if(x->flag ==true || !x->rchild){//没有rchild或者已经访问完了
visit(x);
x = NULL;
}
else{//有rchild且没有被访问
PushStack(S,x);//出栈的重新入栈
x->flag = true;//访问函数置成true
x = x->rchild;
PushStack(S,x); //入栈
}
}
}