二叉树后序遍历算法实现

对于二叉树的三种遍历方式,它们的难易程度是不一样的,最简单的是先序遍历,其次是中序遍历,最难的是后序遍历方式。但是最难的后序遍历方式,却可以通过最简单的先序遍历方式的变形实现,然后把遍历的结果逆序一下就搞定了。哈哈,物极必反啊!

先看一个最简单的后序的遍历方法的实现,利用先序遍历方式的变形,然后逆序

vector<int> PostOrder(TreeNode *root)
{
	vector<int> result;
	stack<const TreeNode*> s;
	const TreeNode* cur = root;

	if(cur != NULL)
		s.push(cur);

	while(!s.empty())
	{
		cur = s.top();
		s.pop();

		result.push_back(cur->val);

		if(cur->left)
			s.push(cur->left);
		if(cur->right)
			s.push(cur->right);
	}

	reverse(result.begin(), result.end());

	return result;
}

  其中用到了一个std中的reverse函数,下面对这个函数做一个简单的说明:

这个函数的头文件为:

#include <algorithm>    // std::reverse

  这个函数对vector逆序的方法就是把vector的begin()和end()作为两个参数传递进去就OK了

 

真正的后续遍历还是比较不容易的,因为除了利用栈之外,还得用一个标记记住当前这个根节点的左右孩子是否已经遍历完了,如果左右都已经遍历完了,那就可以遍历当前的结点了,如果没有遍历完,还必须让该根节点的右孩子和左孩子依次入栈。

使用的标记方法就是:对于当前的根节点,如果它的子树已经遍历完成,在这个当前结点的前一个当前结点,要么是它的左孩子(没有右子树),要么是它的右孩子。所以如果前一个遍历的结点是它的左孩子或者右孩子时,就说明这个根节点的子树都已经遍历完成了。

1.取栈顶元素为当前结点
	如果当前结点没有孩子或者有孩子但是孩子(左或者右)刚刚被访问过
		访问这个当前结点
		更新上一个访问结点为当前的结点
		做一次弹栈操作
	如果有孩子并且孩子(左或者右)都不是上一个被访问的结点
		把右孩子压栈,把左孩子压栈
2.重复1的过程,直到栈为空,结束

  后序遍历代码实现:

vector<int> PostOrder(TreeNode *root)
{
	const TreeNode *cur = NULL;
	const TreeNode *pre = NULL;

	stack<const TreeNode*> s;
	vector<int> result;
	
	cur = root;
	if(cur != NULL)
		s.push(cur);

	while(!s.empty())
	{
		cur = s.top();

		if((cur->left ==NULL && cur->right == NULL) ||
			(pre != NULL && (cur->right == pre || cur->left == pre)))
		{
			result.push_back(cur->val);
			pre = cur;
			s.pop();
		}
		else
		{
			if(cur->right != NULL)
				s.push(cur->right);
			if(cur->left != NULL)
				s.push(cur->left);
		}
	}//while

	return result;
}

  

posted @ 2015-07-25 17:21  stemon  阅读(1055)  评论(0编辑  收藏  举报