剑指offer 题7——重建二叉树

最新推荐文章于 2024-11-13 19:48:54 发布

人间世庄子

最新推荐文章于 2024-11-13 19:48:54 发布

阅读量111

点赞数

分类专栏：剑指offer 文章标签：面试

本文链接：https://blog.csdn.net/rjszz1314/article/details/103319411

版权

剑指offer 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

重建二叉树

题目如下

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如，输入前序遍历{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建的二叉树如下图所示并输出它的头结点指针。

二叉树定义如下：

sturct BinaryTreeNode
{
		int  m_nValue;
		BinaryTreeNode m_pLeft;
		BinaryTreeNode m_pRight;
};

重建的二叉树如下：

由前序遍历的特点我们可以得知，前序遍历的一个元素的值就是root（根节点）的值，而由中序遍历的特点，在root节点的左边是左子树，右边是右子树
在这里插入图片描述
这样就得到了左子树和右子树的前序遍历序列和中序遍历序列。
看到这有点想法了吧，又是前序遍历序列和中序遍历序列，那接下来不就是做一样的事情了嘛，所以就是递归嘛。

代码：

BinaryTreeNode* Construct(int* preoeder,int* inorder,int length)
{
        if(preoeder==nullptr || inorder==nullptr || length<=0)
                return nullptr;

        return ConsturctCore(preoeder,preoeder+length-1,inorder,inorder+length-1);
}

BinaryTreeNod* ConsturctCore(int* startPreorder,int* endPreorder,int* startInorder,int* endInorder)
{
                //前序遍历序列第一个数字就是根节点的值
                int rootValue = startPerorder[0];
                BinaryTreeNode* root=new BinaryTreeNode();
                root->m_nValue=rootValue;
                root->m_pLeft=root->m_pRight=nullptr;

                if(startPreorder==endPreorder)
                {
                        if(startInorder==endInorder && *startPreorder==*endPreorder)  //数组中只要一个元素
                                        return root;
                        else
                                std::out<<"Invalid input"<<std::endl;
                        }

                //在中序遍历中找到根节点的值
                int* rootInorder = startInorder;
                while(rootInorder<=endInorder && *rootInorder!=rootValue) 
                                ++rootInorder;

                if(rootInorder==endInorder && *rootInorder!=rootValue)
                                 std::out<<"Invalid input"<<std::endl;

                int leftLength=rootInorder-startInorder;
                int* leftPreorderEnd=startInorder+leftLength;  

                if(leftLength>0) 
                {
                //构建左子树,传入左子树的前序遍历序列和中序遍历序列
                 root->m_pLeft=ConsturctCore(startInorder+1,leftPreorderEnd,startInorder,rootInorder-1);
                }

                if(leftLength<endPreorder-startPreorder)
                {
                 //构建右子树，传入右子树的前序遍历序列和中序遍历序列                       
                root->m_pRight=ConsturctCore(leftPreorderEnd+1,endPreorder,rootInorder+1,endInorder);
                }

        return root;
}