输入某二叉树的前序遍历和中序遍历结果,重建出二叉树。
对于这个问题,我们首先要明白二叉树的性质,在二叉树的前序遍历中,第一个数子总是树的根节点的值,在中序列中,根节点的值总在序列的中间,而在后序遍历中,根节点的值总是在序列的最后一个。因此我们需要扫描中序遍历序列,才能找到根结点的值,并不断递归得分成左右两部分,下面是实现代码:
#include <cstring>
#include <sys/stat.h>
#include <unistd.h>
#include <stdio.h>
#include <iostream>
using namespace std;
int preorder[] = {1, 2, 4, 7, 3, 5, 6, 8}; //假如这是已知的前序序列
int inorder[] = {4, 7, 2, 1, 5, 3, 8, 6}; //假如这是已知的中序序列
int flag = 0;
struct BSTreeNode
{
int m_nValue; // value of node
BSTreeNode *m_pLeft; //left child of node
BSTreeNode *m_pRight; //right child of node
};
void Construct(BSTreeNode *&Node, int PreLeft, int PreRight, int InLeft, int InRight) //重建二叉树
{
Node = new BSTreeNode();
Node->m_nValue = preorder[PreLeft]; //找前序数列的第一个数
Node->m_pLeft = NULL;
Node->m_pRight = NULL;
int k = InLeft;
while(Node->m_nValue != inorder[k] && k <= InRight)
{
k++;
}
if(k > InRight) //前序序列与中序序列不匹配
{
cout<<"输入不合法"<<endl;
flag = 1;
return ;
}
if(k - InLeft > 0)
{
Construct(Node->m_pLeft, PreLeft+1, PreLeft+k - InLeft+1, InLeft, k-1);
}
if(InRight - k > 0)
{
Construct(Node->m_pRight, PreLeft+k - InLeft+1, PreRight, k+1, InRight);
}
}
void BSTreeOutPut(BSTreeNode* Node)
{
if(Node == NULL)
return ;
// cout << Node->m_nValue << endl; //树的前序输出
BSTreeOutPut(Node->m_pLeft);
// cout << Node->m_nValue << endl; //中序输出
BSTreeOutPut(Node->m_pRight);
cout << Node->m_nValue << endl; //后序输出
}
int main()
{
BSTreeNode *Node;
Construct(Node, 0, 7, 0, 7);
if(!flag)
BSTreeOutPut(Node);
return 0;
}
如果题目要求不必构造出二叉树,让我们找他的后序序列,此时我们可以对序列进行中,右,左的递归,每一次取当前序列中的根节点值(也就是当前序列的第一个值),并存入栈中,根据栈先进后出的特点,就能找出具有左,右,中性质的后序序列。
同样,如果已知中序序列与后序序列,求前序序列,则我们可以对序列进行中,左,右递归,每次取当前序列中的根节点值(也就是当前序列中的最后一个值),并存入队列中,根据队列先进先出的特点,就能找出前序序列