2021-10-21 106. 从中序与后序遍历序列构造二叉树(递归)

注:

题目:
根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如,给出:
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

题解:
思路
首先解决这道题我们需要明确给定一棵二叉树,我们是如何对其进行中序遍历与后序遍历的:

  • 中序遍历的顺序是每次遍历左孩子,再遍历根节点,最后遍历右孩子。
  • 后序遍历的顺序是每次遍历左孩子,再遍历右孩子,最后遍历根节点。

因此根据上文所述,我们可以发现后序遍历的数组最后一个元素代表的即为根节点。知道这个性质后,我们可以利用已知的根节点信息在中序遍历的数组中找到根节点所在的下标,然后根据其将中序遍历的数组分成左右两部分,左边部分即左子树,右边部分为右子树,针对每个部分可以用同样的方法继续递归下去构造。

在这里插入图片描述

复杂度分析
时间复杂度:O(n),其中 n 是树中的节点个数。

空间复杂度:O(n)。我们需要使用 O(n) 的空间存储哈希表,以及 O(h)(其中 h 是树的高度)的空间表示递归时栈空间。这里 h < n,所以总空间复杂度为 O(n)。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    map<int,int> mp;
    TreeNode* design(vector<int>& inorder,vector<int> postorder,int in_left,int in_right,int post_left,int post_right){
        if(in_left>in_right){
            return nullptr;
        }
        //根节点在前序遍历中的位置
        int middleindex=mp[postorder[post_right]];
        //左子树长度
        int lefttreesize=middleindex-in_left;
        TreeNode* root=new TreeNode(inorder[middleindex]); 
        root->left=design(inorder,postorder,in_left,in_left+lefttreesize-1,post_left,post_left+lefttreesize-1);
        root->right=design(inorder,postorder,in_left+lefttreesize+1,in_right,post_left+lefttreesize,post_right-1);
        return root;
    }
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        int size=inorder.size();
        if(size==0){
            return nullptr;
        }
        for(int i=0;i<size;i++){
            mp[inorder[i]]=i;
        }
        return design(inorder,postorder,0,size-1,0,size-1);
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值