题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回
思路:运用递归的思想,每一棵子树的构建方式都一样,先递归的构建左右子树,最后生成大树返回根节点
代码实现:
class Solution {
public:
TreeNode* reConstruct(vector<int> pre,//原始先序数列
int startPre,//当前子树先序在原始先序数列中的起始位置
int endPre,//当前子树先序在原始先序数列中的结束位置
vector<int> vin,//原始中序数列
int startVin,//当前子树中序在原始中序数列中的起始位置
int endVin )//当前子树中序在原始中序数列中的结束位置
{
if(startPre>endPre||startVin>endVin) return NULL;
TreeNode* root=new TreeNode(pre[startPre]);//当前先序的第一个数为当前子树的根节点
for(int i=startVin;i<=endVin;i++)//在当前子树的中序数列中找到根节点下标位置i
{
if(vin[i]==pre[startPre])
{
//先构建当前子树的左子树
root->left=reConstruct(pre,startPre+1,startPre+i-startVin,vin,startVin,i-1);
//在构建当前子树的右子树
root->right=reConstruct(pre,startPre+i-startVin+1,endPre,vin,i+1,endVin);
break;
}
}
//当前树的左右子树构建完成,返回这棵树
return root;
}
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
int Prelength=pre.size();//先序长度
int Vinlength=vin.size();//中序长度
return reConstruct(pre,0,Prelength-1,vin,0,Vinlength-1);
}
};