首页 > 代码库 > 重建二叉树

重建二叉树

题目:输入某个二叉树的前序遍历和中序遍历的结果,重建出该二叉树。假设输入的前序遍历和中序遍历结果中都不包含重复数字。二叉树的结点定义如下:

struct BinaryTreeNode
{
    int m_nValue;
    BinaryTreeNode* m_pLeft;
    BinaryTreeNode* m_pRight;
};


分析:在前序遍历中,第一个数是根节点,在中序遍历中找到这个数,则这个数之前的数就是它的左子树,之后就是它的右子树,根据此,利用递归就可以重建此二叉树。具体实现如下:

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* startPreorder, int* endPreorder,int* startInorder,int* endInorder)
{
    int rootValue=http://www.mamicode.com/startPreorder[0];"Invalid input.");
     }
     
     //在中序遍历中找到根节点的值
     int* rootInorder=startInorder;
     while(rootInorder<=endInorder&&*rootInorder!=rootValue)
         ++rootInorder;
     if(rootInorder==endInorder&&*rootInorder!=rootValue)
         throw std::exception("Invalid input.");
     int leftLength=rootInorder-startInorder;
     int* leftPreorderEnd=startPreorder+leftLength;
     if(leftLength>0)
     {
         //构建左子树
         root->m_pLeft=ConstructCore(startPreorder+1,leftPreorderEnd,startInorder,rootInorder-1);
     }
     if(leftLength<endPreorder-startPreorder)
     {
         root->m_pRight=ConstructCore(leftPreorderEnd+1,endPreorder,rootInorder+1,endInorder);
     }
     
     return root;
 }


本文出自 “仙路千叠惊尘梦” 博客,请务必保留此出处http://secondscript.blog.51cto.com/9370042/1581802

重建二叉树