题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果都不含重复的数字。假如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
问题分析:递归思想处理
前序遍历顺序为:节点->左子树->右子树
中序遍历顺序为:左子树->节点->右子树
在前序遍历数组中,拿到第一个节点数据,然后在中序遍历数组中找到该节点,然后将其数组分为两部分。然后根据两部分数组在重复以上动作分别重建其左子树与右子树。
代码实现:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
if(pre.empty()&&vin.empty())
return NULL;
TreeNode* root = new TreeNode(pre[0]);
int pos = 0;
for(pos;pos<vin.size();pos++)
{
if(pre[0] == vin[pos])
break;
}
vector<int>_preleft,_vinleft,_preright,_vinright;
for(int i = 0;i<pos;i++)
{
_preleft.push_back(pre[i+1]);
_vinleft.push_back(vin[i]);
}
for(int j = pos+1;j<vin.size();j++)
{
_preright.push_back(pre[j]);
_vinright.push_back(vin[j]);
}
root->left = reConstructBinaryTree(_preleft,_vinleft);
root->right = reConstructBinaryTree(_preright,_vinright);
return root;
}