剑指offer-4.重建二叉树(递归)

在这里插入图片描述

解题思路

递归法

根据二叉树的前序遍历和中序遍历,根据前序遍历,可以找到第一个值就是根节点的值,到中序遍历中去寻找根节点,就可以确定左子树和右子树,由此递归进行,构建出完整二叉树。

上面说到需要在前序遍历中拿到根节点的值,然后去中序遍历中找这个值的位置。
所以为了方便期间,用哈希表建立中序遍历中值和下标的映射关系。

递归:

终止条件:
一棵树前序遍历的起点pre_left大于
一棵树前序遍历的终点pre_right就终止递归,返回空节点nullptr;
在这里插入图片描述

根据pre中的起点,即第一个值就是根节点的值,那么我们就可以构建出根节点了。
在这里插入图片描述
根据上面所说的映射关系,找到这个值在中序遍历的位置。
在这里插入图片描述
根据根节点在中序遍历的位置,我们就知道左子树中节点的个数了。
在这里插入图片描述
有了上面的准备工作,咱们就可以递归地去构建左右子树了,重要的就是左右子树前序遍历和中序遍历在pre和vin中的哪个范围上。
在这里插入图片描述
好了,最后返回root就可以了!


完整代码:

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    unordered_map<int,int>m; //记录pre中的节点在vin中的位置,可以通过节点的值直接索引到在vin中的位置,建立映射关系
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        int n=pre.size();
        //特殊情况,直接返回空树
        if(n==0)return nullptr; 
        //为了方便起见,vin建立值和index的映射关系
        for(auto node_val:pre){
            for(int i=0;i<n;++i){
                if(vin[i]==node_val)
                    m[node_val]=i;
            }
        }
        
        return help(pre,vin,0,n-1,0,n-1);
    }
    
    //参数说明
    //pre_left:一棵树前序遍历的起点   pre_right:一棵树前序遍历的终点
    //vin_left:一棵树中序遍历的起点   vin_right:一棵树中序遍历的终点
    TreeNode* help(vector<int> &pre,vector<int> &vin,int pre_left,int pre_right,int vin_left,int vin_right){
        //终止条件
        if(pre_left>pre_right)return nullptr;
        //构造根节点
        TreeNode* root=new TreeNode(pre[pre_left]);
        //记录根节点在pre和vin中的索引
        int pre_root_index=pre_left;
        int vin_root_index=m[pre[pre_left]];
        //记录根节点左子树中节点的个数,为了递归的时候定位下标
        int left_node_count=vin_root_index-vin_left;
        
        //递归构造左右节点
        root->left=help(pre,vin,pre_left+1,pre_left+left_node_count,vin_left,vin_root_index-1);
        root->right=help(pre,vin,pre_left+left_node_count+1,pre_right,vin_root_index+1,vin_right);
        
        return root;
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值