JZ4. 重建二叉树

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

1. 分析

​ 想到二叉树解题模板,先告诉根节点要怎么做,然后将左右子节点丢给递归即可。

本题是要求重建二叉树,我们可由前序遍历知道根结点,然后在中序遍历中找到根结点的值,此值之左全是根结点的左子树,此值之右全是根结点的右子树。然后左子树中前序遍历的第一个值又是当前子树的根节… 完美的递归思路。

实现思路(递归三部曲):

  1. 递归终止条件:当需要重建的前序、中序序列为空,或者二者不相等时候,就不需要构建,直接返回空指针即可;
  2. 需要返回给上一个递归什么?肯定是重构好的左右子树嘛,再联合根节点,就可以了。
  3. 函数实现什么功能?重构二叉树,每一级的任务都是找到左右子树,联合根节点,组成一个二叉树。

实现细节:

  • 需要四个辅助vector来存每次递归之后剩下的前序、中序序列,因为是从中间分隔开的,所有有四个。
  • 需要两个索引下标,因为前序的第一个为根节点,不会被压入进行重构,故需要跳过这个值。对于中序来说,中间某一个值是rootVal,也不会参与后续的重构,故在找到preL和vinL之后,要对vinR的下标加1再操作。
  • 调用递归将preL和vinL重构成root的左子树,将preR和vinR重构成root的右子树。

2. 用C++写出逻辑:

    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.empty() || vin.empty() || pre.size() != vin.size())
            return nullptr;
        int rootVal = pre[0];
        TreeNode* root = new TreeNode(rootVal);
        
        vector<int>preL, preR, vinL, vinR;
        
        int i = 0, j = 1;
        while(vin[i] != rootVal){
            preL.push_back(pre[j++]);
            vinL.push_back(vin[i++]);
        }
        
        i++;
        while(i < pre.size()){
            preR.push_back(pre[j++]);
            vinR.push_back(vin[i++]);
        }
        
        root -> left = reConstructBinaryTree(preL, vinL);
        root -> right = reConstructBinaryTree(preR, vinR);
        
        return root;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值