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;
    }
};

105、106 共同特征：从前序与中序遍历序列构造二叉树、从中序与后序遍历序列构造二叉树

为什么题目不同，但是解法一样，都是根节点定位法 + 深度优先搜索生成左右子树，是有什么共同特征吗？

共同特征是：中序遍历确定节点的左右子树结构，而前序或后序遍历提供根节点的位置信息。

• 后序遍历：最后一个元素是当前子树的根节点
• 前序遍历：第一个元素是当前子树的根节点

后序三步：

• 是否正确找到后序遍历中的根节点。
• 是否正确地在中序遍历中定位到根节点的索引。
• 是否正确地递归构造了左右子树。

构造根节点：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);

前序同理。