给定任意两种遍历结果,重建树的结构(Leetcode105、106、889)

总结

  1. 前序是中左右,中序是左中右,后序是左右中。
  2. 问题的关键在于计算左子树元素个数
  3. 中序计算左子树元素个数(InRootInd-inL)。而对于前序和后序的情况,用后序计算左子树元素个数(postLind-postL+1)。
  4. 对于中序,需要确定中序中根的位置(InRootInd),根的位置通过前序中第一个元素,或后序中最后一个元素(rootval)确定。而对于前序和后序的情况,需要确定后序中左子树的最后一个元素(左子树的根)位置(postLind),这是通过先序中第二个元素(preL+1,也是左子树的根)确定(lval)。
  5. 通过unordered_map优化检索时间。

105. 从前序与中序遍历序列构造二叉树

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n=preorder.size();
        unordered_map<int,int> hm;
        for(int i=0;i<n;i++){
            hm[inorder[i]]=i;
        }
        return build(preorder,0,n-1,0,n-1,hm);
    }

    TreeNode* build(vector<int>& preorder,int preL,int preR,int inL,int inR,unordered_map<int,int>& hm){
        if(preL>preR||inL>inR){
            return nullptr;
        }
        int rootVal=preorder[preL];
        TreeNode* root=new TreeNode(rootVal);
        int InRootInd=hm[rootVal];
        int leftSize=InRootInd-inL;
        root->left=build(preorder,preL+1,preL+leftSize,inL,InRootInd-1,hm);
        root->right=build(preorder,preL+leftSize+1,preR,InRootInd+1,inR,hm);
        return root;
    }

106. 从中序与后序遍历序列构造二叉树

class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        int n=inorder.size();
        unordered_map<int,int> hm;
        for(int i=0;i<n;i++){
            hm[inorder[i]]=i;
        }
        return build(0,n-1,postorder,0,n-1,hm);
    }

    TreeNode* build(int inL,int inR,vector<int>& postorder,int postL,int postR,unordered_map<int,int> &hm){
        if(inL>inR||postL>postR){
            return nullptr;
        }
        TreeNode *root=new TreeNode(postorder[postR]);
        if(inL==inR){
            return root;
        }
        int inRootInd=hm[postorder[postR]];
        int leftSize=inRootInd-inL;
        root->left=build(inL,inRootInd-1,postorder,postL,postL+leftSize-1,hm);
        root->right=build(inRootInd+1,inR,postorder,postL+leftSize,postR-1,hm);

        return root;
    }
};

889. 根据前序和后序遍历构造二叉树

class Solution {
public:
    TreeNode* constructFromPrePost(vector<int>& preorder, vector<int>& postorder) {
        int n=preorder.size();
        if(n==0){
            return nullptr;
        }
        unordered_map<int,int> posthm;
        for(int i=0;i<n;i++){
            posthm[postorder[i]]=i;
        }
        return dfs(preorder,0,n-1,postorder,0,n-1,posthm);
    }
    TreeNode* dfs(vector<int>& preorder,int preL,int preR,vector<int>& postorder,int postL,int postR,unordered_map<int,int>& posthm){
        if(preL>preR||postL>preR){
            return nullptr;
        }
        TreeNode* root=new TreeNode(preorder[preL]);
        if(preL==preR){
            return root;
        }

        int lval=preorder[preL+1];
        int postLind=posthm[lval];
        // 左子树的元素个数
        int leftSize=postLind-postL+1;
        cout<<"lval="<<lval<<"postLind="<<postLind<<"leftsize="<<leftSize<<endl;
        root->left=dfs(preorder,preL+1,preL+leftSize,postorder,postL,postLind,posthm);
        root->right=dfs(preorder,preL+leftSize+1,preR,postorder,postLind+1,postR-1,posthm);
        
        return root;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值