二叉树的非递归遍历

Posted 2022-12-11 suxinpaul

二叉树的建立与遍历(非递归遍历)

二叉树建立的过程总会出现读入回车出错，现在终于找到了解决办法，用 fflush(stdin)  函数，就可以没输入一个字符建立节点，回车

 

 

#include<stdio.h>
#include<stdlib.h>
#define STACK_INIT_SIZE 100
#define STACKINCREMENT  10
#define OK 1
#define OVERFLOW 0
#define ERROR 0
typedef char TElemType;
typedef int Status;
typedef struct BiTNode
	TElemType data;
	struct BiTNode *lchild,*rchild;
BiTNode,*BiTree;
typedef BiTree SElemType;
typedef struct

	SElemType *base;
	SElemType *top;
	int stacksize;
SqStack;
typedef struct

	int *base;
	int *top;
	int stacksize;
flag;
Status InitStackflag(flag &S)
  
	S.base=(int *)malloc(STACK_INIT_SIZE*sizeof(int));
	if(!S.base) return OVERFLOW;
	else
	
		S.top=S.base;
		S.stacksize=STACK_INIT_SIZE;
	

Status Pushflag(flag &S,int e)

	if(S.top-S.base>=S.stacksize)
	
		S.base=(int *)realloc(S.base,(S.stacksize+STACKINCREMENT)*sizeof(int));
		if(!S.base) return OVERFLOW;
		else	
		 
			S.stacksize+=STACKINCREMENT;	
			*(S.top)=e;
			S.top++;		
		
	
	else 
	
		*(S.top)=e;
		S.top++;					
	

int Popflag(flag &S,int &e)

	if(S.top==S.base) return ERROR;
 	e=*(--S.top);
 	return e;

bool EmptyflagStack(flag &S)

	if(S.base==S.top)return true;
	else return false;

/************************************************************/
Status InitStack(SqStack &S)
  
	S.base=(SElemType *)malloc(STACK_INIT_SIZE*sizeof(SElemType));
	if(!S.base) return OVERFLOW;
	else
	
		S.top=S.base;
		S.stacksize=STACK_INIT_SIZE;
	

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 OVERFLOW;
		else	
		 
			S.stacksize+=STACKINCREMENT;	
			*(S.top)=e;
			S.top++;		
		
	
	else 
	
		*(S.top)=e;
		S.top++;					
	

SElemType Pop(SqStack &S,SElemType &e)

	if(S.top==S.base) return ERROR;
 	e=*(--S.top);
 	return e;

bool EmptyStack(SqStack &S)

	if(S.base==S.top)return true;
	else return false;

/*************************************************************************/
Status CreateBiTree(BiTree &T)

	TElemType ch;
	fflush(stdin); 
	scanf("%c",&ch); 
	fflush(stdin); 
	if(ch==' ')T=NULL;
	else		
		if(!(T=(BiTNode*)malloc(sizeof(BiTNode))))return OVERFLOW;
		T->data=ch;
		printf("请输入 %c 的左节点：\\n",T->data);
		CreateBiTree(T->lchild);
		printf("请输入 %c 的右节点：\\n",T->data);
		CreateBiTree(T->rchild);
	

Status PreOrderTraver(BiTree &T,Status(*Visit)(TElemType))//先序遍历 

	SqStack S;
	BiTree P;
	InitStack(S);P=T;
	while(P||!EmptyStack(S))
	
		if(P)
		
			Visit(P->data);
			Push(S,P);			
			P=P->lchild;
		
		else
		
			Pop(S,P);
			P=P->rchild;
		
	

Status InOrderTraverse(BiTree T,Status(*Visit)(TElemType))//中序遍历 

	SqStack S;
	BiTree P;
	InitStack(S);P=T;
	while(P||!EmptyStack(S))
	
		if(P)
		
			Push(S,P);			
			P=P->lchild;
		
		else
		
			Pop(S,P);
			Visit(P->data);
			P=P->rchild;
		
	

Status PostOrderTraverse(BiTree &T,Status(*Visit)(TElemType))//后序遍历 

	SqStack S1;BiTree P;
	flag S2;
	InitStack(S1);P=T;InitStackflag(S2);
	int sign,e;//记录结点从栈中弹出的次数 
	while(P||!EmptyStack(S1))
	
		if(P)
		
			Push(S1,P);//第一次遇到结点T时压入其指针 
			Pushflag(S2,1);//置标志为1 
			P=P->lchild;
		
		else
		
			while(!EmptyflagStack(S2))
			
				sign=Popflag(S2,e);
				Pop(S1,P);
				if(1==sign)//表示走过T的左子树 
				
					Push(S1,P);
					Pushflag(S2,2);
					P=P->rchild;
					break;
				
				else
				
					if(2==sign)//表示T的左右子树都已走过 
					
						Visit(P->data);
						P=NULL;
					
				
			
		
	

Status PrintElement(TElemType e)

	printf("%c ",e);
	return OK;

int main()
	
	BiTree T;
	printf("请输入树根：\\n"); 
	CreateBiTree(T);
	printf("先序遍历：\\n");
	PreOrderTraver(T,PrintElement);
	printf("\\n中序遍历:\\n");
	InOrderTraverse(T,PrintElement);
	printf("\\n后序遍历:\\n");
	PostOrderTraverse(T,PrintElement);
	printf("\\n");