剑指 Offer 07. 重建二叉树

剑指 Offer 07. 重建二叉树

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

例如：

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回的二叉树：
  	3
   / \
  9  20
    /  \
   15   7

• 前序遍历的第一个元素为树的根节点 node 的值；
• 在中序遍历中搜索根节点 node 的索引，可将中序遍历划分为[左子树|根节点|右子树]。
• 根据中序遍历中的左右子树的节点数量，可将前序遍历划分为[根节点|左子树|右子树]。

采用分治算法：

• 建立根节点 node:节点值为 preorder[root]。

• 划分左右子树：查找根节点在中序遍历中的索引i；

• 构建左右子树，开启左右子树递归：

  根节点索引	中序遍历左边界	中序遍历右边界
  左子树（root+1）	left	i-1
  右子树(根节点索引+左子树长度+1)（root+i-left+1）	i+1	right

代码：

class Solution {
private:
    vector<int>preorder;
    unordered_map<int,int>dic;
    TreeNode* recur(int root,int left,int right){
        if(left>right)
            return nullptr;
        TreeNode *node=new TreeNode(preorder[root]);
        int i=dic[preorder[root]];
        node->left=recur(root+1,left,i-1);
        node->right=recuir(root+i-left+1,i+1,right);
        return node;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
		this->preorder=preorder;
		for(int i=0;i<inorder.size();i++)
        {
            dic[inorder[i]]=i;
        }
        return recur(0,0,inorder.size()-1);
    }
};