输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如输入前序遍历序列{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.size() != vin.size() || pre.size() == 0)
return NULL;
int root = pre[0];
//前序的第一个节点为根节点r,在中序中找到该节点r,
//r前的为左子树,r后的为右子树
int i = 0;
while (!vin.empty())
{
if (vin[i] == pre[0])
break;
i++;
}
vector<int> pre_left, pre_right, vin_left, vin_right;
TreeNode* node = new TreeNode(root);
int j = 0;
while (j < i)
{
vin_left.push_back(vin[j]);//将中序的左子树放进数组
pre_left.push_back(pre[j + 1]);//将前序的左子树放进数组
j++;
}
j = i + 1;
while (j < pre.size())
{
vin_right.push_back(vin[j]);//将中序的右子树放进数组
pre_right.push_back(pre[j]);//将前序的右子树放进数组
j++;
}
node->left = reConstructBinaryTree(pre_left, vin_left);
node->right = reConstructBinaryTree(pre_right, vin_right);
return node;
}