【LeetCode】105. 从前序与中序遍历序列构造二叉树(递归、二叉树、前序中序建二叉树)

前序和中序遍历能够唯一确定一颗二叉树，算经典问题吧

题目

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

前序中序唯一确定一棵二叉树

先来复习一下数据结构：
前序遍历：先遍历根节点，在遍历左子树，最后遍历右子树。
中序遍历：先遍历左子树，在遍历根节点，最后遍历右子树。

简单来说：
前序：根左右
中序：左根右

从上面的遍历顺序是不是就发现规律了？

前序遍历的第一个结点是根节点，我们不能判断后面的序列哪些是左子树，哪些是右子树，我们需要确定数量。所以去中序遍历中找到根节点所在的位置，这样能够确定左子树结点个数，这样右子树结点个数也能确定了，再去确定左子树的结构和右子树的结构即可。

时间复杂度O(N)

AC代码

其中，在中序遍历找根节点的位置用哈希表优化查询的时间复杂度从O(N)优化到了O(1)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    //哈希表，记录中序遍历每个元素在中序序列的位置
    unordered_map<int,int> hash;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        //空树
        if (!preorder.size()) return NULL;
        //中序序列插入哈希表
        for (int i = 0 ; i < inorder.size() ; i ++) hash[inorder[i]] = i;
        //建树
        return build(preorder, inorder, 0, preorder.size() -1 , 0 , inorder.size() -1 );
    }

    TreeNode* build(vector<int>& pre, vector<int>& inorder,  int pl, int pr, int inl, int inr){
        //递归终止
        if (pl > pr) return NULL;
        //创建根节点
        auto root = new TreeNode(pre[pl]);
        //根节点在中序遍历的位置
        int k = hash[root->val];
        //建左子树
        root->left = build(pre, inorder, pl + 1, pl + 1 + k - 1 - inl, inl, k -1);
        //建右子树
        root->right = build(pre, inorder, pl + 1 + k - 1 - inl + 1, pr, k + 1, inr);

        return root;
    }
};