重建二叉树

标签：重建二叉树;c++;递归;遍历

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

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=startPreorder[0];
    BinaryTreeNode* root=new BinaryTreeNode();
    root->m_nValue=rootValue;
    root->m_pLeft=root->m_pRight=NULL;
    
    if(startPreorder==endPreorder)
    {
        if(startInorder==endInorder&&*startPreorder==*startInorder)
            return root;
        else
            throw std::exception("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

重建二叉树

标签：重建二叉树;c++;递归;遍历

原文地址：http://secondscript.blog.51cto.com/9370042/1581802