剑指offer. 重建二叉树

题目

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出

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

返回如下的二叉树：

           3    
          / \   
         9  20
           /  \    
          15   7

限制：
0 <= 节点个数 <= 5000

思路

以示例中的数据为例，先序遍历的第一个数字为3，则3为树根，中序遍历中，3左边的数在左子树，3右边的数在右子树。以3建立父节点，再分别根据左子树中的数建立左子树，右子树中的数建立右子树。如此递归下去即可重建二叉树。

代码

class Solution {
private:
    unordered_map<int,int> ump;
    TreeNode* buildTree(const vector<int>& preorder,int pl,int pr,const vector<int>& inorder,int il,int ir){
        if(pl>pr||il>ir) return NULL;//当中序遍历或者先序遍历为空时则是空树，实际上两者是同时空的
        TreeNode* root=new TreeNode(preorder[pl]);
        int posi=ump[preorder[pl]]; //获取中序遍历中树根的位置
        int left_length=posi-il;    //计算左子树的长度
        int right_length=ir-posi;   //计算右子树的长度
        root->left=buildTree(preorder,pl+1,pl+left_length,inorder,il,posi-1); //递归建立左右子树
        root->right=buildTree(preorder,pl+left_length+1,pr,inorder,posi+1,ir); 
        return root;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        for(int i=0;i<inorder.size();i++){ //对中序遍历结果建立起数字到位置的映射加快在中序遍历中寻找根节点的速度
            ump[inorder[i]]=i;
        }
        return buildTree(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
    }
};