题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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;
};
BinaryTreeNode*ConstructCore(int*start_preorder,int*end_preorder,int*start_inorder,int*end_inorder);
//先声明这个核心函数,从参数定义来看,是必须要定位子树对应的前序中序遍历的子序列
BinaryTreeNode*Construct(int*preorder,int*inorder,int length)
{
if(preorder==NULL||inorder==NULL||length<=0)
return NULL;
return ConstructCore(preorder,preorder+length-1,inorder,inorder+length-1);
}
BinaryTreeNode*ConstructCore(int*start_preorder,int*end_preorder,int*start_inorder,int*end_inorder)
{
int rootValue=start_preorder[0];
BinaryTreeNode*root=new BinaryTreeNode();
root->m_nValue=rootValue;
root->m_pLeft=root->m_pRight=NULL;
if(start_preorder==end_preorder)//递归终止条件
{
return root;
}
int*rootinorder=start_inorder;//在中序遍历序列中找到根节点
while(rootinorder<=end_inorder&&rootinorder!=rootValue)
++rootinorder;
int leftLength=rootinorder-start_inorder;
int*left_end_preorder=start_preorder+leftLength;
if(leftLength>0)//如果有左子树
root->m_pLeft=ConstructCore(start_preorder+1,left_end_preorder,start_inorder,rootinorder-1);
if(leftLength<end_preorder-start_preorder)//如果有右子树
root->m_pRight=ConstructCore(left_end_preorder+1,end_preorder,rootinorder+1,end_inorder);
return root;
}