题目:
重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含童复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示的二叉树并输出它的头结点。
二叉树结点的定义如下:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含童复的数字。例如输入前序遍历序列{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 *ReBuiltBiTree(int *preOrder,int *inOrder, int treeNum)
{
if (preOrder==NULL||inOrder==NULL||treeNum<=0)
{
return NULL;
}
//前序遍历的第一个数字是根节点
BinaryTreeNode *root=new BinaryTreeNode;
root->m_nValue=preOrder[0];
root->m_pRight=root->m_pLeft=NULL;
//在中序遍历中找到根结点,则根结点左边结点是左子树,右边结点是右子树
int rootorder=-1;
for (int i=0; i<treeNum; i++)
{
if (root->m_nValue==inOrder[i])
{
rootorder=i;
break;
}
}
if (rootorder==-1)
{
printf("Invalid Input");
}
//重建左子树
int treeNumleft=rootorder;
int *preOrderleft=preOrder+1;
int *inOrderleft=inOrder;
if(treeNumleft>0)
root->m_pLeft=ReBuiltBiTree(preOrderleft,inOrderleft,treeNumleft);
//重建右子树
int treeNumRight=treeNum-treeNumleft-1;
int *preOrderRight=preOrder+treeNumleft+1;
int *inOrderRight=inOrder+treeNumleft+1;
if(treeNumRight>0)
root->m_pRight=ReBuiltBiTree(preOrderRight,inOrderRight,treeNumRight);
return root;
}