通常遍历二叉树是用的递归办法,但是当结点数过多时很容易发生溢出现象。
学习非递归的遍历过程有利于我们更好理解遍历遵循的原理,本程序使用C++的标准模板库中的栈和队列来进行辅助。
//使用C++中的引用构建二叉树,并进行非递归遍历
#include<cstdio>
#include<iostream>
#include<stack>
#include<queue>
#include<cstdlib>
using namespace std;
typedef struct bitNode{
char data;
struct bitNode *lchild, *rchild;
}bitNode,*biTree;
stack<biTree> st;//定义一个存储结点的栈
queue<biTree> q;
//按照先序递归遍历创建一个二叉树
int creatbitree(biTree &T){
char ch;
T = (biTree)malloc(sizeof(bitNode));
if(!T){
exit(0);
}
cin >> ch;
if(ch == '0')
T = NULL;
else{
T->data = ch;
creatbitree(T->lchild);
creatbitree(T->rchild);
}
return 0;
}
//非递归先序遍历
int preorderNoRecursion(bitNode *T){
if(T == NULL){
return 0;
}
st.push(T);
while(!st.empty()){
bitNode *p;
p = st.top();
st.pop();
cout << p->data<< " ";
if(p->rchild != NULL){
st.push(p->rchild);
}
if(p->lchild != NULL){
st.push(p->lchild);
}
}
return 0;
}
//非递归中序遍历
int inorderNoRecursion(bitNode *T){
if(T == NULL)
return 0;
bitNode *p = T;
while(p || !st.empty()){
if(p){
st.push(p);
p = p->lchild;
}
else{
p = st.top();
cout << p->data << " ";
st.pop();
p = p->rchild;
}
}
return 0;
}
//非递归后序遍历
int postorderNoRecursion(bitNode *T){
if(T == NULL)
return 0;
bitNode *p = T;//工作指针
bitNode *pre = NULL;//用来记录前一个访问的结点,判断是否访问完右子树
while(p || !st.empty()){
if(p){
st.push(p);
p = p->lchild;
}
else{
p = st.top();
if(p->rchild != NULL && p->rchild != pre){//还没访问右子树
p = p->rchild;
st.push(p);
p = p->lchild;
}
else{
p = st.top();
cout << p->data << " ";
st.pop();
pre = p;//记录刚刚访问过的结点
p = NULL;
}
}
}
return 0;
}
//使用队列对层次遍历进行辅助
int levelOrder(bitNode *T){
if(T == NULL)
return 0;
bitNode *p;//工作指针
q.push(T);
while(!q.empty()){
p = q.front();
q.pop();
printf("%c ",p->data);
if(p->lchild != NULL)
q.push(p->lchild);
if(p->rchild != NULL)
q.push(p->rchild);
}
return 0;
}
int main(){
bitNode *T;
cout<<"请创建一个二叉树: "<<endl;
creatbitree(T);
cout<<"先序非递归遍历:";
preorderNoRecursion(T);
cout<<endl;
cout<<"中序非递归遍历:";
inorderNoRecursion(T);
cout<<endl;
cout<<"后序非递归遍历:";
postorderNoRecursion(T);
cout<<endl;
cout<<"层次非递归遍历:";
levelOrder(T);
cout<<endl;
return 0;
}
最后运行结果如图