二叉树三种遍历方式的非递归实现

代码说明

• C & C++
• 结构体变量表示节点
• 使用STL中的stack简化操作

#include<cstdio>
#include<iostream>
#include<string.h>
#include<stdlib.h>
#include<stack> //栈
using namespace std;
#define N 100 //最大节点个数

typedef struct mybinarytree
{
	char data;
	struct mybinarytree *left, *right;
}BT;

BT *createBT(char *pre,char *in,int k);//根据先序和中序序列创建二叉树，k表示节点个数
void showTree(BT *T);//以括号的形式显示二叉树

void preVisit(BT *T);//非递归先序遍历
void inVisit(BT *T);//非递归中序遍历
void postVisit(BT *T);//非递归后序遍历


int main(void)
{
	BT *T;
	char pre[]="ABDEFC";
	char in[]= "BEDFAC";
	int k=strlen(pre);

	//创建二叉树
	T=createBT(pre,in,k);
	cout<<"以括号的形式显示二叉树:\n";
	showTree(T);

	cout<<"\n\n先序遍历:\n";
	preVisit(T);
	cout<<"\n\n中序遍历:\n";
	inVisit(T);
	cout<<"\n\n后序遍历:\n";
	postVisit(T);
	cout<<endl<<endl;

	return 0;
}

BT *createBT(char *pre,char *in,int k)//根据先序和中序序列创建二叉树，k表示节点个数
{
	int i,pos;//确定根节点在中序序列中的位置
	char data;
	BT *p;

	if(k==0)
		return NULL;
	else{
		data=pre[0];
		//生成根节点
		p=(BT *)malloc(sizeof(BT));
		p->data=data;
		//确定data在中序序列中的位置
		for(i=0;i<k && in[i]!=data;i++);
		pos=i;
		//生成左子树
		p->left=createBT(pre+1,in,pos);
		//生成右子树
		p->right=createBT(pre+pos+1,in+pos+1,k-pos-1);
		return p;
	}
}

void showTree(BT *T)//以括号的形式显示二叉树
{
	if(T)
	{
		cout<<T->data;
		if(T->left || T->right) //如果有孩子
		{
			cout<<"(";
			showTree(T->left);
			cout<<",";
			showTree(T->right);
			cout<<")";
		}
	}
}

void preVisit(BT *T)//非递归先序遍历
{
	stack<BT *> t; //建立栈
	BT *p;

	if(!T)
	{
		cout<<"栈为空.\n";
		return;
	}

	//初始化
	t.push(T);
	while(!t.empty())//若栈非空
	{
		p=t.top();
		cout<<p->data;
		t.pop();//弹栈
				//将孩子按照先右后左的顺序入栈
		if(p->right)
			t.push(p->right);
		if(p->left)
			t.push(p->left);
	}
}

void inVisit(BT *T)//非递归中序遍历
{
	stack<BT *> t;
	BT *p,*q;

	q=T;//待中序遍历的树的根节点
	while(!t.empty()||q)
	{
		while (q)//一直向左走
		{
			t.push(q);
			q=q->left;
		}
		//弹栈(此时栈非空)
		p=t.top();//表示左子树已经遍历过的根节点
		cout<<p->data;//访问根节点
		t.pop();
		q=p->right;//遍历右子树
	}
}

void postVisit(BT *T)//非递归后序遍历
{
	stack<BT *> t;
	BT *p,*q,*pre;

	q=T; //待后序遍历的子树的根节点
	do{
		while(q)//一直向左走
		{
			t.push(q);
			q=q->left;
		}
		pre=NULL;//最新弹出的节点
		while(!t.empty())
		{
			p=t.top();
			if(p->right==pre)
			{
				cout<<p->data;
				t.pop();
				pre=p;
			}
			else{
				q=p->right;
				break;
			}
		}
	}while(!t.empty());
}

运行结果