已知二叉树的中序和后序(前序)序列可以唯一确定一颗二叉树,例如,
中序序列:1 2 3 4 5 6
后序序列:3 4 2 6 5 1
中序序列:1 2 3 4 5 6
后序序列:3 4 2 6 5 1
可以唯一确定一颗二叉树,如下图:
思路是这样的:
我们知道对于一颗二叉树,根节点是后序遍历序列的最后一个,找到根节点后,我们由中序遍历的特点知道,中序序列中,根节点 左边的是左子树,右边的是右子树。之后分别在左子树、右子树中运用这种思路,这样递归下去,二叉树就唯一确定了。
比如上面的例子,根节点是 1, 则左子树序列为3 2 4, 右子树序列是 6 5,
左子树序列中, 根节点是2,左子树序列是3,右子树序列是4, 右子树序列6 5也可以这样去确定.
下面是代码实现:
// 由中序序列、后序序列得到前序序列
//2017.10.26
#include <stdio.h>
#include <stdlib.h>
#include <string>
#define MAXLENGTH 20
int postStack[] = {3, 4, 2, 6, 5, 1}; // 后序序列
int inStack[] = {3, 2, 4, 1, 6, 5}; // 中序序列
void get_pre_order(int root, int start, int end){
// 三个参数分别是根节点index,中序序列首个和最后一个元素的index
if(start > end) // 此时元素位置都已经确定了,递归结束
return;
int i = start;
while(i < end && inStack[i] != postStack[root]) i++;
// 在中序序列中找到根节点下标i, 左边的是左子树,右边是右子树
printf("%d ", postStack[root]); // 要输出前序序列, 每次先输出一棵树的根节点
get_pre_order(root-(end-i)-1, start, i-1); // 遍历左子树,对第一个参数, 左子树根节点为根节点先减去右子树(end-i),再减一
get_pre_order(root-1, i+1, end);// 遍历右子树,根节点是原先根节点减一,把上图后序遍历一下就知道了
}
int main(void){
get_pre_order(5, 0, 5);
return 0;
}
// 已知中序序列和前序序列得到后序序列也是这个思路,有兴趣的可以试着写一下,下面是参考代码:
void get_post_order(int root, int start, int end){
if(start > end)
return;
int i = start;
while(i < end && inStack[i] != preStack[root]) i++;
get_post_order(root+1, start, i-1);
get_post_order(root+(i-start)+1, i+1, end);
printf("%d ", postStack[root]);
}
// end