题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路分析:
在二叉树的前序遍历序列中,第一个数字总是树的根节点的值。但在中序遍历序列中,根节点的值在序列的中间,左子树的节点的值在根节点的左边,右子树的节点的值在根节点的右边。因此,我们扫描中序遍历序列,先找到根节点的值,然后根据根节点所处的位置,找到中序遍历序列的左子树和右子树。根据中序遍历序列的左子树和右子树,得到前序遍历序列的左子树和右子树。根据左右子树的前序和中序遍历序列,采用递归的方法,将整颗二叉树重建出来。
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
int vin_size=vin.size();
if(vin.size()==0){
return NULL;
}
int rootValue=pre[0];//得到根节点
vector<int> pre_left, pre_right, vin_left, vin_right;
//在中序遍历结果中搜寻根节点,根节点前为vin_left,后为vin_right
int p=0;
for(p; p<vin_size; p++){
if(vin[p]==rootValue){
break;
}
}
for(int i=0; i<vin_size; i++){
if(i<p){
vin_left.push_back(vin[i]);
pre_left.push_back(pre[i+1]);
}
if(i>p){
vin_right.push_back(vin[i]);
pre_right.push_back(pre[i]);
}
}
TreeNode* node = new TreeNode(rootValue);
node->left=reConstructBinaryTree(pre_left, vin_left);
node->right=reConstructBinaryTree(pre_right, vin_right);
return node;
}