问题重述
题目:输入二叉树的前序和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果都不含重复的数字。例如输入前序序列为{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示的二叉树。
思路解析:
解析本题的主要技巧就是要精确的找出顶节点,并找出左右两颗子树的所有元素,然后递归的进行二叉树的构建,具体的实现见代码部分。
代码实现:
struct binaryTreeNode
{
int mData;
binaryTreeNode * mLeft;
binaryTreeNode * mRight;
};
//根据前序和中序序列重建二叉树
binaryTreeNode *createBt(int * pre, int * mid, int l1, int r1, int l2, int r2)
{
if (l1 > r1 || l2 > r2)
{
return nullptr;
}
binaryTreeNode *s = new binaryTreeNode();
s->mLeft = s->mRight = nullptr;
int i = 0;
for (i = l2; i < r2; ++i)
{
if (pre[l1] == mid[i])
{
break;
}
}
s->mData = mid[i];
//对左右子树元素的下标重新进行判断,重构二叉树
s->mLeft = createBt(pre, mid, l1 + 1, l1 + i - l2, l2, i - 1);
s->mRight = createBt(pre, mid, l1+i-l2+1,r1,i+1,r2);
return s;
}
//二叉树的前序遍历和中序遍历,用于验证构建的二叉树是否正确
void preOrder(binaryTreeNode *s)
{
if (s == nullptr)
{
return;
}
printf("%d ", s->mData);
preOrder(s->mLeft);
preOrder(s->mRight);
}
void inOrder(binaryTreeNode *s)
{
if (s == nullptr)
{
return;
}
inOrder(s->mLeft);
printf("%d ", s->mData);
inOrder(s->mRight);
}
void destory(binaryTreeNode *s)
{
if (s != nullptr)
{
binaryTreeNode *lChild = s->mLeft;
binaryTreeNode *rChild = s->mRight;
delete s;
destory(lChild);
destory(rChild);
}
}
测试:
int main()
{
int pre[] = { 1, 2, 4, 7, 3, 5, 6, 8 };
int mid[] = { 4, 7, 2, 1, 5, 3, 8, 6 };
binaryTreeNode *s = new binaryTreeNode();
s = createBt(pre, mid, 0, 7, 0, 7);
preOrder(s);
printf("\n");
inOrder(s);
printf("\n");
destory(s);
return 0;
}
所得结果:将重建后的二叉树的先序后中序的遍历序列打印出来,发现和输入的一样,结果正确。