题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
1. 分析
想到二叉树解题模板,先告诉根节点要怎么做,然后将左右子节点丢给递归即可。
本题是要求重建二叉树,我们可由前序遍历知道根结点,然后在中序遍历中找到根结点的值,此值之左全是根结点的左子树,此值之右全是根结点的右子树。然后左子树中前序遍历的第一个值又是当前子树的根节… 完美的递归思路。
实现思路(递归三部曲):
- 递归终止条件:当需要重建的前序、中序序列为空,或者二者不相等时候,就不需要构建,直接返回空指针即可;
- 需要返回给上一个递归什么?肯定是重构好的左右子树嘛,再联合根节点,就可以了。
- 函数实现什么功能?重构二叉树,每一级的任务都是找到左右子树,联合根节点,组成一个二叉树。
实现细节:
- 需要四个辅助vector来存每次递归之后剩下的前序、中序序列,因为是从中间分隔开的,所有有四个。
- 需要两个索引下标,因为前序的第一个为根节点,不会被压入进行重构,故需要跳过这个值。对于中序来说,中间某一个值是rootVal,也不会参与后续的重构,故在找到preL和vinL之后,要对vinR的下标加1再操作。
- 调用递归将preL和vinL重构成root的左子树,将preR和vinR重构成root的右子树。
2. 用C++写出逻辑:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
if(pre.empty() || vin.empty() || pre.size() != vin.size())
return nullptr;
int rootVal = pre[0];
TreeNode* root = new TreeNode(rootVal);
vector<int>preL, preR, vinL, vinR;
int i = 0, j = 1;
while(vin[i] != rootVal){
preL.push_back(pre[j++]);
vinL.push_back(vin[i++]);
}
i++;
while(i < pre.size()){
preR.push_back(pre[j++]);
vinR.push_back(vin[i++]);
}
root -> left = reConstructBinaryTree(preL, vinL);
root -> right = reConstructBinaryTree(preR, vinR);
return root;
}