二叉树中序遍历方法实现

对于二叉树的遍历,先序的方式是比较简单的,但是中序和后序的方式还是有点麻烦的,这里先给出一个用C++stack的遍历方式:

1.如果当前结点不为空
	把当前结点压入栈
	p=p->left转向其左孩子
2.如果当前结点为空(证明这半棵子树已经遍历完成,需要从栈顶找到树根)
	取栈顶元素为当前结点,栈做一次弹栈操作
	访问当前结点
	p=p->right转向其右孩子
3.重复1、2操作,直到栈为空并且当前结点为NULL,结束。

  算法代码实现:

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

	cur = root;

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

			cur = cur->right;
		}

	}//while

	return result;
}

  这种利用栈的遍历方法很简单,在之前我也设计了一个自己的遍历的方法:

1.如果当前结点为空(NULL)
	取栈顶元素为当前结点,栈做一次弹栈操作
	访问当前结点
	p=p->right转向其右孩子
2.如果当前结点不为空
	如果当前结点有左孩子
		把当前结点压入栈
		p=p->left转向其左孩子
	如果当前结点没有左孩子
		访问当前结点
		p=p->right转向其右孩子
3.重复1、2操作,直到栈为空并且当前结点为NULL,结束。

  

代码实现:

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

	cur = root;

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

   }//while

   return result;
}

  

上面介绍了两种利用栈对树中序遍历的方法,下面还是说一下Morris的遍历方法

1.如果当前结点有左子树
	找到左子树的最右叶子结点
		如果最右叶子结点的右指针域为空(NULL)
			使最右叶子结点的右指针域指向当前结点
			p=p->left转向其左子树
		如果最右叶子结点的右指针域已经指向了当前结点
			访问当前结点
			断开Morris链接
			p=p->right转向其子树
2.如果当前结点没有左子树
	访问当前结点
	p=p->right转向其子树
3.重复1、2操作,直到当前结点为NULL,结束。

  算法代码实现:

vector<int> InOrder(TreeNode *root)
{
	TreeNode *cur = NULL;
	TreeNode *temp = NULL;
	cur = root;
	vector<int> result;

	while(cur != NULL)
	{
		if(cur->left != NULL)
		{
			temp = cur->left;
			while(temp->right != NULL && temp->right != cur)
			{
				temp = temp->right;
			}

			if(temp->right == NULL)
			{
				temp->right = cur;
				cur = cur->left;
			}
			else	//temp->right == cur
			{
				result.push_back(cur->val);
				temp->right = NULL;
				cur = cur->right;
			}
		}
		else	//没有左子树
		{
			result.push_back(cur->val);
			cur = cur->right;
		}
	}//while

	return result;
}

  

posted @ 2015-07-25 14:46  stemon  阅读(477)  评论(0编辑  收藏  举报