题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
- 必备知识点:前序遍历(根左右), 中序遍历(左根右),前序遍历第一个数1必是根节点,对应到中序遍历中根节点的位置,左边的数位于左子树,右边的数位于右子树;可以看出,1节点同时拥有左右子树;然后, 前序遍历中第二个数字2作为节点,对应到中序遍历中,左边为左子树,同时,右边没有数,所以2节点没有右子树。所以,可以通过递归实现二叉树重建。
TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin) {
if (pre.empty() == NULL || vin.empty() == NULL)
return NULL;
size_t index = 0;
for (; index < vin.size(); index++)
{
if (vin[index] == pre[0])
break;
}
TreeNode* root = new TreeNode(pre[0]);
vector<int> pre_left, vin_left;
vector<int> pre_right, vin_right;
for (size_t j = 0; j < index; j++)
{
pre_left.push_back(pre[j + 1]);
vin_left.push_back(vin[j]);
}
for (size_t j = index + 1; j < vin.size(); j++)
{
pre_right.push_back(pre[j]);
vin_right.push_back(vin[j]);
}
root->left = reConstructBinaryTree(pre_left, vin_left);
root->right = reConstructBinaryTree(pre_right, vin_right);
return root;
}