重建二叉树

题目

思路：先根据二叉树的的前序遍历确定根节点，然后再根据中序遍历确定根节点的左右子树进行建树。

在中序数组中寻找根节点的时候需要O(n)遍历才能得到，可以先用一个哈希表存储起来，之后只要O(1)即可找到根节点在中序数组中的位置。

class Solution {
public:
    /* 
    核心思想就是根据前序确定根节点，
    然后根据根节点和中序遍历确定左右分支，
    进行递归
     */
    unordered_map<int, int> m;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        for(int i = 0;i<inorder.size();i++){
            m[inorder[i]] = i;
        }
        return build(preorder, 0, 0, inorder.size()-1);
    }

    /* 
    @ param index根节点的下标
    @ param l 子树的左边界
    @ param r 子树的右边界
     */
    TreeNode* build(vector<int>& preorder, int index, int l, int r){
        if(l > r) return NULL;
        //先创建根节点
        TreeNode* node = new TreeNode( preorder[index] );
        //找出根节点的在中序数组中的下标
        int root = m[preorder[index]];
              
        node->left = build(preorder, index+1, l, root-1);
        
        //前序根左右,因此右子树根节点的下标是 当前根节点+左子树节点数+1 
        node->right = build(preorder, index + (root - l)+1, root+1, r);
        return node;
    }
};