已知先序中序,构造二叉树

给定一棵树的前序遍历 preorder 与中序遍历  inorder。请构造二叉树并返回其根节点

法一:递归分治

对于任意一颗树而言,前序遍历的形式总是


[ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]
即根节点总是前序遍历中的第一个节点。而中序遍历的形式总是


[ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]
只要我们在中序遍历中定位到根节点,那么我们就可以分别知道左子树和右子树中的节点数目。由于同一颗子树的前序遍历和中序遍历的长度显然是相同的,因此我们就可以对应到前序遍历的结果中,对上述形式中的所有左右括号进行定位。

而中序序列的左右括号很容易定位,根据上一个左右括号和本次的节点位置就可以。

这样以来,我们就知道了左子树的前序遍历和中序遍历结果,以及右子树的前序遍历和中序遍历结果,我们就可以递归地对构造出左子树和右子树,再将这两颗子树接到根节点的左右位置。

struct TreeNode{
int val;
TreeNode* left;
TreeNode* right;
TreeNode (a):left=nuullptr,right=nuullptr{
val=a;
}
}
 

 
class Solution {
    private: unordered_map<int,int> inorder_map;
public:
 TreeNode* building(vector<int>& preorder,vector<int>& inorder,int preorder_left,int preorder_right,int inorder_left,int inorder_right){
     //以前序遍历为基准,右超过左代表子树不存在了
         if(preorder_left>preorder_right)return nullptr;
        //记录本次先序遍历元素位置
        int preorder_root_index= preorder_left;
        //记录本次中序遍历元素位置
        int inorder_root_index=inorder_map[preorder[preorder_root_index]];
        //构造节点
        TreeNode* node=new TreeNode(preorder[preorder_root_index]);
        //求出左子树的大小,知道等会如何安排左子树先序序列的左右界 和右子树先序序列的左右界。
        int left_subtree_size=inorder_root_index-inorder_left;
        //根据基本原理,定位先序序列中,左子树左右中括号,以及中序序列中左子树左右中括号
        node->left=building(preorder,inorder,preorder_left+1,preorder_left+left_subtree_size,inorder_left,inorder_root_index-1);
          //根据基本原理,定位先序序列中,右子树左右中括号,以及中序序列中右子树左右中括号
         node->right=building(preorder,inorder,preorder_left+left_subtree_size+1,preorder_right,inorder_root_index+1,inorder_right);
         return node;
    }
    
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n=inorder.size();
        //构造哈希映射,有利于在常数时间内,定位中序遍历序列中元素
        for(int i=0;i<n;i++){
            inorder_map[inorder[i]]=i;
        }
        //building函数递归的构造节点
        return building(preorder,inorder,0,n-1,0,n-1);
    }
   
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值