利用栈实现的二叉树的先、中、后序遍历

最新推荐文章于 2024-05-29 16:57:33 发布

原创最新推荐文章于 2024-05-29 16:57:33 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #二叉树 #栈 #遍历

数据结构专栏收录该内容

9 篇文章

订阅专栏

本文介绍如何使用栈来实现二叉树的先序、中序和后序遍历。通过栈的操作，如压栈、弹栈等，实现对二叉树节点的有效访问。在每种遍历方式中，通过特定的逻辑处理来确保正确的访问顺序。

利用栈实现二叉树的遍历

栈的操作参考：

栈、循环队列的基本操作

先序遍历

status PreOrderbyStack(BiTree *T, void (*visit)(BiTree *T))
{
	stack s;
	BiTree *pmove;
	initStack(&s);

	pmove = T;
	Push(&s, T);
	while(!IsStackEmpty(&s))
	{
		Pop(&s, &pmove);
		if(pmove == NULL)
		{
			continue;
		}
		else
		{
			visit(pmove);
			Push(&s, pmove->rchild);
			Push(&s, pmove->lchild);
		}
	}
	destroyStack(&s);
	return TRUE;
}

中序遍历

status InOrderbyStack(BiTree *T, void (*visit)(BiTree *T))
{
	stack s;
	BiTree *pmove;
	BiTree *flag;
	initStack(&s);

	pmove = T;
	Push(&s, T);
	Push(&s, (BiTree *)1);//加入标号，表示这个节点左孩子还未访问
	while(!IsStackEmpty(&s))
	{
		Pop(&s, &flag);
		Pop(&s, &pmove);
		if((int)flag == 1)
		{
			while(pmove->lchild != NULL)
			{
				Push(&s, pmove);
				Push(&s, (BiTree *)0);//加入标号，表示这个节点左孩子已被访问
				pmove = pmove->lchild;
			}
		}
		visit(pmove);
		if(pmove->rchild != NULL)
		{
			Push(&s, pmove->rchild);
			Push(&s, (BiTree *)1);//加入标号，表示这个节点左孩子还未访问
		}
	}

	destroyStack(&s);
	return TRUE;
}

后序遍历

status PostOrderbyStack(BiTree *T, void (*visit)(BiTree *T))
{
	stack s;
	BiTree *pmove;
	BiTree *flag;
	initStack(&s);

	pmove = T;
	Push(&s, T);
	Push(&s, (BiTree *)0);//加入标号，表示这个节点左孩子还未访问
	while(!IsStackEmpty(&s))
	{
		Pop(&s, &flag);
		Pop(&s, &pmove);
		if((int)flag == 0)
		{
			while(pmove->lchild != NULL)
			{
				Push(&s, pmove);
				Push(&s, (BiTree *)1);//加入标号，表示这个节点左孩子已被访问
				pmove = pmove->lchild;
			}
		}
		if((int)flag < 2)
		{
			if(pmove->rchild != NULL)
			{
				Push(&s, pmove);
				Push(&s, (BiTree *)2);//加入标号，表示这个节点右孩子已被访问
				pmove = pmove->rchild;

				Push(&s, pmove);
				Push(&s, (BiTree *)0);//加入标号，表示这个节点左孩子还未访问
				continue;
			}
		}
		visit(pmove);
	}

	destroyStack(&s);
	return TRUE;
}