首页 > 其他 > 详细

重建二叉树

时间:2014-11-24 13:49:13      阅读:246      评论:0      收藏:0      [点我收藏+]

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

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

重建二叉树

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!