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

Posted 2023-03-07 Debroon

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

• • 题目
  • 算法设计：深度优先搜索

 

---

题目

传送门：https://leetcode.cn/problems/construct-binary-tree-from-inorder-and-postorder-traversal/

 

---

算法设计：深度优先搜索

构造二叉树，第一件事一定是找根节点，再构造左右子树。

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

    TreeNode* build(vector<int>& in, vector<int>& post, int inStart, int inEnd, int& postIdx) 
        if(inStart > inEnd) return nullptr;
        TreeNode* root = new TreeNode(post[postIdx--]);             // 构造根节点
        int inIdx = mp[root -> val];                                // 获取中序遍历中根节点的位置
        root -> right = build(in, post, inIdx+1, inEnd, postIdx);   // 构造右子树区间
        root -> left  = build(in, post, inStart, inIdx-1, postIdx); // 构造左子树区间
        return root;
    
;