剑值offer66题之每日一题——第四题

YXR11111

于 2018-10-05 17:58:13 发布

阅读量148

点赞数

CC 4.0 BY-SA版权

分类专栏：手撕代码

本文链接：https://blog.youkuaiyun.com/qq_41598851/article/details/82945808

手撕代码专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种利用前序和中序遍历结果重建二叉树的递归算法。通过解析给定序列，构建出完整的二叉树结构，并详细解释了代码实现过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

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

思路：运用递归的思想，每一棵子树的构建方式都一样，先递归的构建左右子树，最后生成大树返回根节点

代码实现：

class Solution {
public:
    TreeNode* reConstruct(vector<int> pre,//原始先序数列
                          int startPre,//当前子树先序在原始先序数列中的起始位置
                          int endPre,//当前子树先序在原始先序数列中的结束位置
                          vector<int> vin,//原始中序数列
                          int startVin,//当前子树中序在原始中序数列中的起始位置
                           int endVin )//当前子树中序在原始中序数列中的结束位置
    {
        if(startPre>endPre||startVin>endVin) return NULL;
        TreeNode* root=new TreeNode(pre[startPre]);//当前先序的第一个数为当前子树的根节点
    
        for(int i=startVin;i<=endVin;i++)//在当前子树的中序数列中找到根节点下标位置i
        {
            if(vin[i]==pre[startPre])
            {
                //先构建当前子树的左子树
                root->left=reConstruct(pre,startPre+1,startPre+i-startVin,vin,startVin,i-1);
                //在构建当前子树的右子树
                root->right=reConstruct(pre,startPre+i-startVin+1,endPre,vin,i+1,endVin);
                break;
            }
        }
        //当前树的左右子树构建完成，返回这棵树
        return root;        
    }
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
      int Prelength=pre.size();//先序长度
      int Vinlength=vin.size();//中序长度
      return reConstruct(pre,0,Prelength-1,vin,0,Vinlength-1);
    }
};