题目:重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含童复的数字。例如输入前序遍历序列{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; root->m_pLeft=ReBuiltBiTree(preOrderleft,inOrderleft,treeNumleft); //重建右子树 int treeNumRight=treeNum-treeNumleft-1; int *preOrderRight=preOrder+treeNumleft+1; int *inOrderRight=inOrder+treeNumleft+1; root->m_pRight=ReBuiltBiTree(preOrderRight,inOrderRight,treeNumRight); return root; }
时间: 2024-10-18 05:33:23