在前面的文章里,都是用递归实现输出二叉树的操作,那么非递归又是如何实现呢?
(案例)构建二叉树:
下面附上代码,由于里面的注释已经很详细了,也作为自己记录复习,有兴趣的小伙伴可以看看哦。
注:由于注释已经很详细了,所以这里就没有过多的文字说明了。(以下代码幅值可直接运行哦~)
#include <iostream>
#include <stdlib.h>
#define ElemType char
#define maxSize 100
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
using namespace std;
typedef struct Tree{
ElemType data;
struct Tree *lchild;
struct Tree *rchild;
}Tree;
void createTree(Tree * &root){
ElemType c;
scanf("%c",&c);
if(c == ' '){
return;
}
root = (Tree*)malloc(sizeof(Tree));
root->data = c;
root->lchild = NULL;
root->rchild = NULL;
createTree(root->lchild);
createTree(root->rchild);
}
/* 先序遍历非递归算法*/
void frontPrint(Tree *p){
cout<< "前序遍历:";
//初始化栈
Tree *stack[maxSize];
int top;
top = -1;
stack[++top] = p;
//如果栈不为空
while(top != -1){
p = stack[top--];
//先访问根节点
cout<<p->data<<" ";
// 如果右孩子不为空 保留右孩子
if(p->rchild != NULL)
stack[++top] = p->rchild;
// 如果左孩子不为空 保留左孩子
if(p->lchild != NULL)
stack[++top] = p->lchild;
/*
这里解释下为什么是先保留右孩子再保留左孩子。
因为栈的特点是:先进后出,现在先序遍历的顺序是 根->左->右
那么输出根节点之后我们想先输出左孩子节点,那么就先让 右孩子入栈,
再让左孩子入栈,这样等会出栈的时候左孩子就先比右孩子先出栈,
故就可以达到先访问左孩子的节点了。
*/
}
cout<<endl;
}
/* 中序遍历非递归算法*/
void centerPrint(Tree *p){
cout<< "中序遍历:";
Tree *stack[maxSize];
int top;
top = -1;
// 如果当前指针不为空 或者栈内还有元素
while(p != NULL || top!= -1){
// 一直递归到当前节点的最左孩子
while(p != NULL){
// 这里只入栈根孩子节点就行。
// 为什么不需要入栈左孩子呢?
// 其实入栈左孩子与根节点是同时发生的!入栈根节点(A)后,
// 下一层循环会入栈(B)。
// 显然 入栈(B)既是(A)的左孩子又是(C)的根节点
// 所以说入栈的既是左节点,又是根节点。 即同时发生的
// 如下图:
// A
// /
// B
// /
// C
// 而为什么不需要入栈右孩子呢?
// 因为我们既然都入栈了根(左)节点
// 那么出栈该节点不就可以访问右孩子节点。
stack[++top] = p;
p = p->lchild;
}
if(top != -1 ){
// 出栈
p = stack[top--];
// 访问左节点(因为上一个循环结束代表最左节点无左孩子!)
// 同时若该节点存在右孩子的话,也就相当于访问了根节点,
cout<<p->data<<" ";
//访问它的右孩子
p = p->rchild;
// 所以从代码执行逻辑而言,我们得到的访问顺序也必为:左->根->右
}else{
p = NULL;
}
}
cout<<endl;
}
/*
后序遍历需要借助两个栈
栈1作用:按照 根->右->左 的顺序把 树的各个节点压入栈2中
栈2作用:出栈即为后序遍历。
*/
void realPrint(Tree *p){
cout<< "后序遍历:";
if(p != NULL){
Tree *stack1[maxSize];
Tree *stack2[maxSize];
int top1,top2;
top1 = top2 = -1;
stack1[++top1] = p;
// 如果栈1不为空
while(top1 != -1){
//栈1退栈
p = stack1[top1--];
//把退栈的元素压入栈二中,此时相当于压入根节点。
stack2[++top2] = p;
//如果当前节点的左孩子不为空,那么压入栈1里
if( p->lchild !=NULL){
stack1[++top1] = p->lchild;
}
//如果当前节点的右孩子不为空,那么压入栈1里
if( p->rchild !=NULL){
stack1[++top1] = p->rchild;
}
// 想想为什么是 先压入 左孩子 然后是右孩子呢?
// 后序遍历的顺序为:左->右->根
// 而栈的特点是 先进后出。
// 所以先压入左孩子 再压入右孩子,这样就可以保证
// 在下一轮回循环里 右孩子先退栈,然后把它的左右孩子压入栈2里!
// 最后是左孩子退栈,把它的左后孩子再压入栈2里。
// 结合栈的特点,就能知道 左孩子节点时先于右孩子节点访问的。
}
while(top2 != -1){
// 此时栈2已经是按照根->右->左的顺序压栈的,
// 所以退栈顺序刚好相反 故不用做调整。
p = stack2[top2--];
cout<<p->data<<" ";
}
}
cout<<endl;
}
int main(int argc, char** argv) {
Tree *root;
cout<<"请按照前序构建二叉树:";
createTree(root);
frontPrint(root);
centerPrint(root);
realPrint(root);
return 0;
}