整理大二数据结构代码之二叉树的遍历

#include<iostream>
#include<cstdlib>
using namespace std;

#define BiTree ElemType
#define MAXSIZE 10
#define INCREMENT 10

enum Status{OK,ERROR};

//-----二叉树的二叉链表存储表示--------
typedef struct BiTNode{
	char data;
	struct BiTNode *lchild,*rchild;//左右孩子指针
}BiTNode,*BiTree;

//-----顺序栈的存储表示以及基本操作部分-----
typedef struct SqStack{
    ElemType *top,*base;
	int stacksize;
}SqStack;

void InitSqStack(SqStack &S){//创建一个空栈
    S.base=(ElemType*)malloc(MAXSIZE*sizeof(ElemType));
	if(!S.base)exit(0);
	S.top=S.base;
}

bool EmptyStack(SqStack S){//判断栈是否为空，若是则返回true，否则返回false
    if(S.base==S.top){
		//cout<<"栈为空"<<endl;
		return true;
	}
	return false;
}

int StackLength(SqStack S){//返回栈的长度
    return S.top-S.base;
}

void GetTop(SqStack S,ElemType &e){
    e=*(--S.top);
}

void Push(SqStack &S,ElemType e){//在栈顶插入数据e
    if(StackLength(S)>=MAXSIZE){
	    S.base=(ElemType*)realloc(S.base,(S.stacksize+INCREMENT)*sizeof(ElemType));
		if(!S.base)exit(0);
		S.stacksize+=INCREMENT;
	}
	*(S.top++)=e;
}

void Pop(SqStack &S,ElemType &e){//删除栈顶元素，并用e返回其值
    if(EmptyStack(S))exit(0);
	e=*(--S.top);
}

//-------二叉树的基本操作部分---------

void InitBiTree(BiTree &T){//按先序构造二叉链表表示二叉树T
	char ch;
	cout<<"请输入:";
	cin>>ch;
	if(ch=='#')T=NULL;//
	else{
		T=new BiTNode;
		T->data=ch;
		InitBiTree(T->lchild);
		InitBiTree(T->rchild);
	}
}

Status PrintElement(char e){//输出
	cout<<e<<" ";
	return OK;
}

void PreOredrTraverse1(BiTree T){//先序遍历二叉树（递归算法）
	if(T){
		PrintElement(T->data);
		PreOredrTraverse1(T->lchild);
		PreOredrTraverse1(T->rchild);

	}
}//PreOrderTraverse

void InOrderTraverse1(BiTree T){//中序遍历二叉树（递归算法）
	if(T){
		InOrderTraverse1(T->lchild);
		PrintElement(T->data);
		InOrderTraverse1(T->rchild);
	}
}

void PostOrderTraverse(BiTree T){//后序遍历二叉树（递归算法）
	if(T){
		PostOrderTraverse(T->lchild);
		PostOrderTraverse(T->rchild);
		PrintElement(T->data);
	}
}

void PreOredrTraverse2(BiTree T){//先序遍历二叉树（非递归算法）
    SqStack S;
	InitSqStack(S);
	Push(S,T);
	BiTree p;
	p=T;
	while(!EmptyStack(S)){
	    Pop(S,p);
		while(p){
		    cout<<p->data<<" ";
			if(p->rchild)Push(S,p->rchild);
			p=p->lchild;
		}
	}
}

void InOrderTraverse2(BiTree T,SqStack S){//中序遍历二叉树（非递归算法）
    Push(S,T);
	BiTree p=new BiTNode;
	p=T;
	while(!EmptyStack(S)){
	    while(GetTop(S,p),p)Push(S,p->lchild);
		Pop(S,p);
		if(!EmptyStack(S)){
		    Pop(S,p);
			cout<<p->data<<" ";
			Push(S,p->rchild);
		}
	}
}

int main(int argc, char* argv[])//主函数测试
{
	BiTree T;
	InitBiTree(T);
	PreOredrTraverse1(T);//先序遍历（递归算法）
	cout<<endl;
	InOrderTraverse1(T); //中序遍历（递归算法）
	cout<<endl;
	PostOrderTraverse(T); //后序遍历（递归算法）
	SqStack S;
	InitSqStack(S);
	cout<<endl;
	PreOredrTraverse2(T); //先序遍历（非递归算法）
	cout<<endl;
        InOrderTraverse2(T,S);// 中序遍历（非递归算法）
	cout<<endl;
	return 0;
}