
/**
* 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>ans;
TreeNode* dfs(vector<int>& pre, vector<int>& in,int pl,int pr,int il,int ir)//pl是指当前pl根节点在前序的范围,后面是在中序 //的范围
{
if(pl>pr)return nullptr;//不存在范围说明为空
int rec=pre[pl];//当前根的元素值
int k=ans[rec];//根在中序遍历的下标
int length=k-il;//根左孩子范围
auto root=new TreeNode(rec);
root->left=dfs(pre,in,pl+1,pl+length,il,k-1);//pl+length为pl左树的上界,k-1为中序的上界;
root->right=dfs(pre,in,pl+1+length,pr,k+1,ir);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int n=inorder.size();
for(int i=0;i<n;i++)ans[inorder[i]]=i;//把中序下标存入哈希表
return dfs(preorder,inorder,0,n-1,0,n-1);
}
};