重建二叉树并非递归前序/中序/后序遍历

《编程之美》3.9重建二叉树的题目:根据中序遍历和前序遍历结果重建二叉树

#include <iostream>
#include <cassert>
#include <algorithm>
#include <stack>

using namespace std ; 

struct NODE 
{
	NODE *pLeft;
	NODE *pRight;
	char chValue;
};

void Rebuild(char *preOrder, char *inOrder, int nTreeLen, NODE **pRoot)
{
	if(preOrder==NULL&&inOrder==NULL)
	{
		*pRoot = NULL;
		return;
	}

	if (nTreeLen == 0)
	{
		return;
	}
	

	if (*pRoot==NULL)
	{
		*pRoot = new NODE;
		assert(*pRoot!=NULL);
		(*pRoot)->pLeft=NULL;
		(*pRoot)->pRight = NULL;
		(*pRoot)->chValue = *preOrder;
	}

	if (nTreeLen==1)
	{
		return;
	}

	char *preTmp = preOrder;
	char *inTmp = inOrder;
	int count=0;
	while(*inTmp!=*preTmp && count<nTreeLen)
	{
		if (inTmp==NULL)	//这个地方要注意判断一下,不过如果假定输入串没有错误的话,这里一般不会出问题。
		{
			return;
		}
		inTmp++;
		count++;
	}

	Rebuild(preOrder+1,inOrder,count,&((*pRoot)->pLeft));
	Rebuild(preOrder+count+1,inOrder+count+1,nTreeLen-count-1,&((*pRoot)->pRight));

}


void PreOrderTraversalNonrecursive(NODE *proot)
{
	stack<NODE*> tStack;

	if (proot==NULL)
	{
		return;
	}

	tStack.push(proot);

	while(!tStack.empty())
	{
		NODE *pnode = tStack.top();
		cout << pnode->chValue << " ";

		tStack.pop();

		if (pnode->pRight!=NULL)
		{
			tStack.push(pnode->pRight);
		}

		if (pnode->pLeft!=NULL)
		{
			tStack.push(pnode->pLeft);
		}
	}

	cout << endl;
}

void InOrderTraversalNonrecursive(NODE *proot)
{
	if (proot==NULL)
	{
		return;
	}

	stack<NODE *> tstack;
	NODE *pnode = proot;
	
	while(!tstack.empty() || pnode!=NULL)
	{
		if (pnode!=NULL)
		{
			tstack.push(pnode);
			pnode=pnode->pLeft;
		}
		else
		{
			pnode = tstack.top();
			cout << pnode->chValue << " ";
			tstack.pop();
			pnode = pnode->pRight;
		}
	}

	cout << endl;

}

void PostOrderTraversalNonrecursive(NODE *proot)
{
	if (proot==NULL)
	{
		return;
	}

	stack<NODE*> s1, s2;
	NODE* pnode = proot;

	s1.push(pnode);
	while(!s1.empty())
	{
		pnode = s1.top();
		s1.pop();
		if (pnode->pLeft!=NULL)
		{
			s1.push(pnode->pLeft);
		}
		if (pnode->pRight!=NULL)
		{
			s1.push(pnode->pRight);
		}

		s2.push(pnode);
	}

	while(!s2.empty())
	{
		pnode = s2.top();
		cout<<pnode->chValue<<" ";
		s2.pop();
	}

	cout<<endl;
}
int main()
{

	char *preStr= "abdcef";
	char *inStr = "dbaecf";

	NODE *pRoot = NULL;

	Rebuild(preStr,inStr,6,&pRoot);

	PreOrderTraversalNonrecursive(pRoot);
	InOrderTraversalNonrecursive(pRoot);
	PostOrderTraversalNonrecursive(pRoot);

	system("pause");

	return 0;

}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值