二叉树的遍历

最新推荐文章于 2024-01-27 15:36:29 发布

WHY1990607

最新推荐文章于 2024-01-27 15:36:29 发布

阅读量118

点赞数

本文链接：https://blog.csdn.net/WHY1990607/article/details/83959638

版权

//bintraverse.h

#include "malloc.h"
#include "stdio.h"
#include <iostream>
using namespace std;
#define STACK_INIT_SIZE 20
#define STACKINCREMENT 5
#define ERROR 0
#define OK 1
typedef int ElemType; //结点元素类型为整型！！
typedef int Status;
typedef struct BinaryTree
{
int flag;
ElemType data;
struct BinaryTree *lchild;
struct BinaryTree *rchild;
}*BiTree,BiTNode,*SElemType;

//
typedef struct{
   SElemType *base;
   SElemType *top;
   int stacksize;
}SqStack;

Status InitStack(SqStack &S)
{
   S.base=(SElemType *)malloc(STACK_INIT_SIZE*sizeof(SElemType));
   if(!S.base) return ERROR;
   S.top=S.base;
   S.stacksize=STACK_INIT_SIZE;
   return OK;
}

Status GetTop(SqStack S,SElemType &e)
{
   if(S.top==S.base) return ERROR;
   e=*(S.top-1);
   return OK;
}

Status Push(SqStack &S,SElemType e)
{
   if(S.top-S.base>=S.stacksize)
   {
       S.base=(SElemType *)realloc(S.base,
           (S.stacksize+STACKINCREMENT)*sizeof(SElemType));
       if(!S.base) return ERROR;
       S.top=S.base+S.stacksize;
       S.stacksize+=STACKINCREMENT;
   }
   *S.top++=e;
   return OK;
}

Status Pop(SqStack &S,SElemType &e)
{
   if(S.top==S.base)return ERROR;
   e=*--S.top;
   return OK;
}

Status StackEmpty(SqStack S)
{
if(S.top==S.base) return OK;
else return ERROR;
}

Status CreateBiTree(BiTree &T)
{//先序创建用二叉链表表示的二叉树，0表示空树
int tdata;
cin>>tdata;
if (tdata==0) T=NULL; //用0表示空树！！
else
{
if(!(T=new BiTNode)) return ERROR;
T->data=tdata;
CreateBiTree(T->lchild);
CreateBiTree(T->rchild);
}
return OK;
}

int printelem(ElemType d)
{
printf("%d\t",d);
return OK;
}

Status PreOrderTraverse(BiTree T,Status (*Visit)(ElemType e))//先序遍历二叉树T的非递归算法，对每个数据元素调用函数Visit.
{
SElemType e;
SqStack s;
InitStack(s); Push(s, T);
while(T||!StackEmpty(s))
{
if(T)
{
if(!Visit(T->data)) return ERROR;
if(T->lchild)
Push(s,T->lchild);
T=T->lchild;
}

else
{
Pop(s,e);
if(GetTop(s,T)&&T)
{
T=T->rchild;
Pop(s,e);
Push(s,T);
}
}
}
return OK;
}
//3
Status InOrderTraverse1(BiTree T, Status (*Visit)(ElemType e)) {
// 算法6.2
// 采用二叉链表存储结构，Visit是对数据元素操作的应用函数。
// 中序遍历二叉树T的非递归算法，对每个数据元素调用函数Visit。
SqStack S;
BiTree p;
InitStack(S); Push(S, T); // 根指针进栈
while (!StackEmpty(S))
{
while (GetTop(S, p) && p) Push(S, p->lchild); // 向左走到尽头
Pop(S, p); // 空指针退栈
if (!StackEmpty(S))
{ // 访问结点，向右一步
Pop(S, p);
if (!Visit(p->data)) return ERROR;
Push(S, p->rchild);
}
}
return OK;
} // InOrderTraverse1

Status PostOrderTraverse(BiTree T,int (*Visit)(ElemType d))
{
if(T)
{
if(PostOrderTraverse(T->lchild,Visit))
if(PostOrderTraverse(T->rchild,Visit))
if(Visit(T->data)) return OK;
return ERROR;
}
else return OK;
}
Status PostOrderTraverse1(BiTree T,int (*Visit)(ElemType d))//后序遍历二叉树的非递归算法，对每个数据元素调用Visit
{
SElemType e;
SqStack s;
InitStack(s); Push(s,T);
while(T||!StackEmpty(s))
{
if(T)
{
if(T->lchild)
Push(s,T->lchild);
T=T->lchild;
}
else
{
Pop(s,e);
Visit(e->data);
if(GetTop(s,T)&&T->flag!=1)
{
if(T->rchild)
Push(s,T->rchild);
T->flag=1;
T=T->rchild;
}
else
{
Pop(s,e);
Visit(e->data);
if(GetTop(s,e)&&e->flag==1)
{
Visit(e->data);
Pop(s,e);
}
T=NULL;
}
}
}
return OK;
}