任务:给定一颗有n个结点的二叉树,结点的编号是1,2,3,...,n。已知二叉树结点编号的后序和中序列表,试设计一个算法,确定该二叉树结点编号的前序列表。
输入:第1行有1个正整数n,表示给定的二叉树有n个结点。接下来的两行中,第1行是二叉树结点的后序遍历列表,第2行是二叉树结点的中序遍历列表
输出:二叉树的前序遍历列表
测试输入:
5
3 4 5 2 1
3 4 1 5 2
测试输出:
1 4 3 2 5
这是一个结论,由二叉树的前序/后序列表+中序列表可以唯一确定一颗二叉树,其它组合则不行。这里应该要到的是分治思想。再加上二叉树中序遍历顺序为(左子树->根节点->右子树),后序遍历顺序为(左子树->右子树->根节点)
由二叉树的后序遍历可知,遍历结果的最后一个结点必然是这棵树的根节点,再利用这个根节点的值去中序遍历中找到这个结点。那么此时中序遍历序列中,该结点左侧即为该树的左子树,右侧即为该树的右子树。此时输出该节点,再依照上面的方法,对左子树进行递归,直至后序遍历序列只剩一个结点为止,然后再去遍历右子树。这样恰好就完成了这棵树的先序遍历(根节点->左子树->右子树)
#include <iostream>
using namespace std;
// post存储后序遍历序列,pl是左边界,pr是右边界
// in 存储中序遍历序列,il是左边界,ir是右边界
void PrintPreOrder(int post[], int pl, int pr, int in[], int il, int ir)
{
if( pl > pr )
return;
int root = post[pr];
cout << root << " ";
if( pl == pr )
return;
// 寻找中序遍历中的根节点
int i;
for(i = il; i <= ir; ++i)
if( in[i] == root )
break;
// 划分
PrintPreOrder(post, pl, pl+i-il-1, in, il, i-1);//左子树
PrintPreOrder(post, pl+i-il, pr-1, in, i+1, ir);//右子树
}
int main()
{
int i, n;
while( cin >> n )
{
int post[n+1], in[n+1]; //后序,中序序列
for(i = 1; i <= n; ++i)
cin >> post[i];
for(i = 1; i <= n; ++i)
cin >> in[i];
PrintPreOrder(post, 1, n, in, 1, n);
cout << endl << "-----END-----" << endl;
}
return 0;
}
// For test
// 12
// 8 9 4 10 5 2 11 6 12 7 3 1
// 8 4 9 2 5 10 1 11 6 3 7 12
// Result should be
// 1 2 4 8 9 5 10 3 6 11 7 12