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

/**
 * 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 {
private:
    TreeNode* traversal (vector<int>& inorder, vector<int>& postorder) {
        if (postorder.size() == 0) return NULL;

        // 后序遍历数组最后一个元素，就是当前的中间节点
        int rValue = postorder[postorder.size() - 1];
        TreeNode* root = new TreeNode(rValue);

        // 叶子节点
        if (postorder.size() == 1) return root;

        // 找到中序遍历的切割点
        int dIndex;
        for (dIndex = 0; dIndex < inorder.size(); dIndex++) {
            if (inorder[dIndex] == rValue) break;
        }

        // 切割中序数组
        // 左闭右开区间：[0, delimiterIndex)
        vector<int> lInorder(inorder.begin(), inorder.begin() + dIndex);
        // [delimiterIndex + 1, end)
        vector<int> rInorder(inorder.begin() + dIndex + 1, inorder.end() );

        // postorder 舍弃末尾元素
        postorder.resize(postorder.size() - 1);

        // 切割后序数组
        // 依然左闭右开，注意这里使用了左中序数组大小作为切割点
        // [0, leftInorder.size)
        vector<int> lPostorder(postorder.begin(), postorder.begin() + lInorder.size());
        // [leftInorder.size(), end)
        vector<int> rPostorder(postorder.begin() + lInorder.size(), postorder.end());

        root->left = traversal(lInorder, lPostorder);
        root->right = traversal(rInorder, rPostorder);

        return root;
    }
public:
/*第一步：如果数组大小为零的话，说明是空节点了。第二步：如果不为空，那么取后序数组最后一个元素作为节点元素。
第三步：找到后序数组最后一个元素在中序数组的位置，作为切割点。第四步：切割中序数组，切成中序左数组和中序右数组 （顺序别搞反了，一定是先切中序数组）.第五步：切割后序数组，切成后序左数组和后序右数组.第六步：递归处理左区间和右区间*/
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if (inorder.size() == 0 || postorder.size() == 0) return NULL;
        return traversal(inorder, postorder);
    }
};