【剑指offer】2.3.4 树

面试题7：重建二叉树

题目：输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如：输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建如下图所示的二叉树并输出它的头节点。二叉树节点的定义如下：

struct BinaryTreeNode

{

    int m_nValue;

    BinaryTreeNode *m_pLeft;

    BinaryTreeNode *m_pRight;

};

解答：在二叉树的前序遍历过程中，第一个数字总是树的根节点的值。但在中序遍历序列中，根节点的值在序列的中间，左子树的节点的值位于根节点的值的左边，而右子树的节点的值位于根节点的值的右边。如图所示，前序遍历序列的第一个数字1就是根节点的值。扫描中序遍历序列，就能确定根节点的值的位置。根据中序遍历的特点，在根节点的值1前面的3个数字都是左子树的值，位于1后面的数字都是右子树节点的值。 

    由于在中序遍历序列中，有3个数字是左子树节点的值，因此左子树共有3个左子节点。同样，在前序遍历序列中，根节点后面的3个数字就是3个左子树节点的值，再后面的所有数字都是右子树节点的值。这样就在先序遍历和中序遍历两个序列中分别找到了左、右子树对应的子序列。

    既然已经分别找到了左、右子树的先序遍历序列和中序遍历序列，就可以用同样的方法分别构建左、右子树。也就是说，接下来的事可以用递归的方法去完成。代码如下：

TreeNode* reConstructTree(vector<int> pre,int abegin,int aend,vector<int> vin,int bbegin,int bend)
{
	if(abegin >= aend || bbegin >= bend)
	{
		return NULL;
	}

	TreeNode *s = new TreeNode(pre[abegin]);

	int pos = bbegin;
	for(;pos < bend;pos++)
	{
		if(vin[pos] == pre[abegin])
		{
			break;
		}
	}

	s->left = reConstructTree(pre,abegin+1,abegin+pos-bbegin+1,vin,bbegin,pos);
	s->right = reConstructTree(pre,abegin+pos-bbegin+1,aend,vin,pos+1,bend);

	return s;
}
  
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
{
	if(pre.size() == 0 || vin.size() == 0)
	{
		return NULL;
	}

	return reConstructTree(pre,0,pre.size(),vin,0,vin.size());
}

面试题8：二叉树的下一个节点

题目：给定一棵二叉树和其中的一个节点，如何找出中序遍历序列的下一个节点？树中的节点除了有两个分别指向左、右子节点的指针，还有一个指向父节点的指针。

解答：在下图中的二叉树的中序遍历序列是{d,b,h,e,i,a,f,c,g}。下面将以这棵树为例来分析如何找出二叉树的下一个节点。

    如果一个节点有右子树，那么它的下一个节点就是它的右子树中的最左子节点。也就是说，从右子节点出发一直沿着指向左子节点的指针，就能找到它的下一个节点。例如，上图中节点b的下一个节点是h，节点a的下一个节点是f。

    接着分析一个节点没有右子树的情形。如果节点是它父节点的左子节点，那么它的下一个节点就是它的父节点。例如，上图中节点d的下一个节点是b，节点f的下一个节点是c。

    如果一个节点既没有右子树，而且它还是它父节点的右子节点，那么就沿着指向父节点的指针一直向上遍历，直到找到一个是它父节点的左子节点的节点。如果这样的节点存在，那么这个节点的父节点就是要找的下一个节点。

    为了找到上图中节点i的下一个节点，可以沿着指向父节点的指针向上遍历，先到达节点e。由于节点e是它父节点b的右节点，则继续向上遍历到达节点b。节点b是它父节点a的左子节点，因此节点b的父节点a就是节点i的下一个节点。

    找出节点g的下一个节点的步骤类似。先沿着指向父节点的指针到达节点c，由于节点c是它父节点a的右子节点，继续向上遍历到达节点a。由于节点a是树的根节点，它没有父节点，因此节点g没有下一个节点。代码如下：

TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
	if(NULL == pNode)
	{
		return NULL;
	}

	TreeLinkNode *pNext = NULL;
	if(pNode->right != NULL)
	{
		TreeLinkNode *pRight = pNode->right;
		while(NULL != pRight->left)
		{
			pRight = pRight->left;
		}
		pNext = pRight;
	}
	else if(NULL != pNode->next)
	{
		TreeLinkNode *pParent = pNode->next;
		while(NULL != pParent && pNode == pParent->right)
		{
			pNode = pParent;
			pParent = pParent->next;
		}

		pNext = pParent;
	}

	return pNext;
}