leetcode之从前序与中序遍历序列构造二叉树（C++）

参考链接

1. https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

题目描述

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

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

解题思路

前序遍历的第一个节点，就是树的根节点；这个根节点在中序遍历中，左边的就是左子树，右边的就是右子树；如此分别对左子树和右子树递归即可。

为了更方便地确定由前序遍历确定的根节点在中序遍历中的位置，我们构建一个节点值–>中序遍历位置的哈希表，题目的注意事项恰好允许我们这样做。

代码

class Solution {
private:
    unordered_map<int, int> inMap;
public:
    TreeNode* myBuildTree(vector<int>& preorder, int preStart, int preEnd, vector<int>& inorder, int inStart, int inEnd)
    {
        if(preStart > preEnd) return nullptr;

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

        int inRoot = inMap[preorder[preStart]];
        int numsLeft = inRoot - inStart;

        root->left = myBuildTree(preorder, preStart + 1, preStart + numsLeft, inorder, inStart, inRoot - 1);
        root->right = myBuildTree(preorder, preStart + numsLeft + 1, preEnd, inorder, inRoot + 1, inEnd);
        return root;
    }

    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n = preorder.size();
        for(int i = 0; i < n; i ++)
        {
            inMap[inorder[i]] = i;
        }
        return myBuildTree(preorder, 0, n - 1, inorder, 0, n - 1);
    }

};