LeetCode106 : 从中序与后序遍历序列构造二叉树

1. 题目描述

---

---

2. 思路分析

---

---

3. 代码展示

class Solution {
public:
    TreeNode* _build(vector<int>& inorder, int L, int R, vector<int>& postorder, int& idx) {
        //左区间 > 右区间, 说明没有元素, 是空树
        if(L > R)
            return nullptr;
        //创建当前结点
        TreeNode* root = new TreeNode(postorder[idx]);
        //在中序序列中查找当前结点的位置
        int  i = L;
        for(; i <= R; i++) {
            //找到之后直接跳出循环
            if(root->val == inorder[i])
                break;
        }

        //注意后序遍历反过来应该先创建右子树
        if(R > i) {
            //说明右子树存在, 向下递归创建右子树
            //区间: [i+1, R]
            --idx;
           root->right = _build(inorder, i + 1, R, postorder, idx);
        }
        else {
            //右子树不存在
            root->right = nullptr;
        }

        if(L < i) {
            //说明左子树存在, 向下递归创建左子树
            //区间: [L, i - 1]
            --idx;
            root->left = _build(inorder, L, i - 1, postorder, idx);
        }
        else {
            root->left = nullptr;
        }

        return root;
    }

    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        int idx =  postorder.size() - 1;
        return _build(inorder, 0, inorder.size() - 1, postorder,idx);
    }
};