LeetCode 889. 根据前序和后序遍历构造二叉树

题目描述

889. 根据前序和后序遍历构造二叉树

解法：

首先通过数据结构都知道，前序加后序并不能唯一确定一棵二叉树，但是不要紧，我们依然还是能够通过递归构造出一棵可能的二叉树，具体流程如下

1. 首先把前序遍历结果的第一个元素或者后序遍历结果的最后一个元素确定为根节点的值
2. 然后把前序遍历结果的第二个元素作为左子树的根节点的值（实际上，如果根节点的左子树有可能是空指针，那么此时的这个元素就应该是右子树的根节点，所以导致了二叉树结构不唯一，如下图所示）
3. 在后序遍历结果中寻找左子树根节点的值，从而确定了左子树的索引边界，进而确定右子树的索引边界，递归构造左右子树即可

例如，$\text{preorder}=[1,2,3],\text{postorder}=[3,2,1]$

我们这里给出一幅图说一下关于索引的确定

/**
 * 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:
    TreeNode* constructFromPrePost(vector<int>& preorder, vector<int>& postorder) {
        return helper(preorder, 0, preorder.size() - 1, postorder, 0, postorder.size() - 1);

    }
    TreeNode* helper(vector<int>& preorder, int preStart, int preEnd, vector<int>& postorder, int postStart, int postEnd){
        if (preStart > preEnd) return nullptr;

        if (preStart == preEnd)
        {
            TreeNode* tmp = new TreeNode(preorder[preStart]);
            return tmp;
        }

        int root_val = preorder[preStart];
        TreeNode* root = new TreeNode(root_val);

        int root_left_val = preorder[preStart + 1];
        int index = -1;
        for (int i = postStart; i <= postEnd; i++)
        {
            if (postorder[i] == root_left_val)
            {
                index = i;
                break;
            }
        }

        int leftSize = index - postStart + 1;
        root->left = helper(preorder, preStart + 1, preStart + leftSize,
                            postorder, postStart, index);
        root->right = helper(preorder, preStart + leftSize + 1, preEnd,
                            postorder, index + 1, postEnd - 1);
        return root;
    }
};