题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
考查的知识点:树、dfs、数组
首先,我们要知道树的遍历有以下三种:
前序遍历:根->左->右
中序遍历:左->根->右
后序遍历:左->右->根
已知前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
前序遍历的第一个元素1必为根节点,再从中序遍历中找到1,则可以把中序遍历分为左右两半,分别为左子树和右子树,不断重复这个步骤,直到所有的节点加到树上。
我们用一个for循环,遍历中序遍历中的元素,每当找到根节点,就根据这个元素在中序遍历中的位置i,将其分别添加到左右子树,其长度为i-vin_left
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
return constructBinaryTree(pre,0,pre.size()-1,vin,0,vin.size()-1);
}
TreeNode* constructBinaryTree(vector<int>pre,int b1,int e1,vector<int>vin,int b2,int e2)
{
if(b1>e1) return nullptr;
TreeNode* root=new TreeNode(pre[b1]);
for(int i=b2;i<=e2;i++)
{
if(vin[i]==root->val)
{
root->left=constructBinaryTree(pre,b1+1,b1+i-b2,vin,b2,i-1);//i-b2是左子树的长度
root->right=constructBinaryTree(pre,b1+i-b2+1,e1,vin,i+1,e2);
break;
}
}
return root;
}