问题描述:
输入一棵二叉树前序遍历和中序遍历的结果,请重建该二叉树。
注意:
- 二叉树中每个节点的值都互不相同;
- 输入的前序遍历和中序遍历一定合法;
样例:
解决问题的关键是如何根据中序遍历和前序遍历还原出二叉树的结构,学习数据结构的时候,这样类似的题目我们也是做过的。前序遍历第一个节点是root节点,之后根据该节点找到在中序遍历中的位置,那么该位置左右分别是这个二叉树的左右子树。然后依次遍历这个二叉树的左右子树。难点是确定二叉树的左右子树范围。找到根节点root的位置 k 以后,对于中序遍历而言,右子树是从 k 后的节点开始到 ir 结束,左子树是从 il 到k 前面的节点,对于前序遍历而言,左子树是从 pl+1 开始的,到(pl+1)+(k-1-il)=pl+k-il 中序遍历的左子树长度与前序遍历左子树长度(k-1-il)相等可以得到该等式。右子树就可想而知是pl+k-il+1 到 pr。
为啥前序遍历的右边界值是preorder.size()-1,便于下面创建root节点。
class Solution {
public:
map<int,int>hash;
vector<int> preorder,inorder;
TreeNode* buildTree(vector<int>& _preorder, vector<int>& _inorder) {
preorder=_preorder,inorder=_inorder;
for(int i=0;i<inorder.size();i++) hash[inorder[i]]=i;
return dfs(0,preorder.size()-1,0,inorder.size());
}
TreeNode* dfs(int pl,int pr,int il,int ir)
{
if(pl>pr) return nullptr;
auto root=new TreeNode(preorder[pl]);
int k=hash[root->val];
auto left=dfs(pl+1,pl+k-il,il,k-1);
auto right=dfs(pl+k-il+1,pr,k+1,ir);
root->left=left,root->right=right;
return root;
}
};