二叉树的非递归遍历C语言实现

腾讯面试中被问到二叉树的非递归遍历实现,当时记得不太清楚,回来专门复习了非递归的实现,整理代码如下:

//采用二叉链表存储方式的二叉树,非递归中序遍历C语言实现代码
#include
#include 
//函数结果状态代码
#define TRUE 1
#define FALSE 0
#define OK 1
#define ERROR 0
#define INFEASIBLE -1
#define OVERFLOW -2
//Status是函数的类型,其值是函数结果状态代码
typedef int Status;

#define STACK_INIT_SIZE 10
#define STACKINCREMENT 2

typedef struct BiTNode{
	int data;
	struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;

typedef struct{
	BiTree *base;
	BiTree *top;
	int stacksize;
}SqStack;

Status InitStack(SqStack *S)
{
	/* 构造一个空栈S */
	(*S).base = (BiTree*)malloc(STACK_INIT_SIZE*sizeof(BiTree));
	if(!(*S).base)
		exit(OVERFLOW);      /* 存储分配失败 */
	(*S).top = (*S).base;
	(*S).stacksize = STACK_INIT_SIZE;
	return OK;
}

Status StackEmpty(SqStack S)
{
/* 若栈S为空栈,则返回TRUE,否则返回FALSE */
	if(S.top==S.base)
		return TRUE;
	else
		return FALSE;
}
int StackLength(SqStack S)
{
	/* 返回S的元素个数,即栈的长度 */
	return S.top-S.base;
}

Status GetTop(SqStack S,BiTree *e)
{
	/* 若栈不空,则用e返回S的栈顶元素,并返回OK;否则返回ERROR */
	if(S.top>S.base)
	{
		*e = *(S.top-1);
		return OK;
	}
	else
		return ERROR;
}
Status Push(SqStack *S,BiTree e)
{
	if((*S).top-(*S).base>=(*S).stacksize)/*栈满,追加存储空间*/
	{
		(*S).base = (BiTree *)realloc((*S).base,((*S).stacksize+STACKINCREMENT)*sizeof(BiTree));
		if(!(*S).base)
			exit(OVERFLOW);
		(*S).top = (*S).base + (*S).stacksize;
		(*S).stacksize += STACKINCREMENT;
	}
	*((*S).top)++=e;
	return OK;
}
Status Pop(SqStack *S,BiTree *e)
{
	/* 若栈不空,则删除S的栈顶元素,用e返回其值,并返回OK;否则返回ERROR */
   if((*S).top==(*S).base)
	   return ERROR;
   *e=*--(*S).top;
   return OK;
}

/*按前序输入二叉树中结点的值*/
/*#表示空树,构造二叉链表表示二叉树T*/
void CreateBiTree(BiTree *T) //输入前序序列1,2,3方法为1 2 0 0 3 0 0
{
	int data;
	scanf("%d",&data);
	if(data == 0)
		*T = NULL;
	else
	{
		*T = (BiTree)malloc(sizeof(BiTNode));
		if(!*T)
			exit(OVERFLOW);
		(*T)->data = data;
		CreateBiTree(&(*T)->lchild);
		CreateBiTree(&(*T)->rchild);
	}
}

Status Visit(int e)
{
	printf("%d ",e);
	return OK;
}
//中序非弟归遍历二叉树(二叉链表存储)
Status InOrderTraverse(BiTree T,Status (*Visit)(int)){
	SqStack S;
	BiTree p = T;
	InitStack(&S);
	while(p || !StackEmpty(S)){
		if(p){
			Push(&S,p);
			p = p->lchild;
		}
		else
		{
			Pop(&S,&p);
			if(!Visit(p->data))
				return ERROR;
			p = p->rchild;
		}
	}
	return OK;
}

void main()
{
	BiTree T;
	CreateBiTree(&T);
	InOrderTraverse(T,Visit);
}




你可能感兴趣的:(数据结构与算法学习)