上次记录了广义表生成二叉树的过程,我们也可以通过前序和中序,或者中序和后序,来确定和构建一棵唯一的二叉树。
还是同样的图,它的前序,中序,后序遍历序列分别是:
pre: ABCDEF
in: CBDAEF
post: CDBFEA
以下是通过前序和中序构建二叉树的过程:
- 获取前序字符串的第一个字符A,它作为当前根节点,然后扫描中序字符串,找到A的位置,创建根节点存储结构。
- 然后在中序字符串中确定左子树中序为CBD,再去前序字符串中截取相同长度的子串,确定左子树前序为BCD,有了左子树的前序和中序,就可以递归调用创建左子树,跟当前根节点关联起来。
- 同样地,我们可以确定右子树的前序为EF,中序为EF,然后递归调用创建右子树,跟当前根节点关联起来。
- 整个过程是递归的,直到前序和中序字符串只有一个字符时,直接创建叶子节点并跟父节点关联起来。
通过中序和后序构建二叉树跟上述过程类似,由后序确定当前根节点后,再通过中序确定左子树和右子树序列,然后递归创建左子树和右子树,并与当前根节点关联。
下面是实现代码:
JS版:
//二叉树节点结构
function BinTreeNode(data) {
this.data = data;
this.leftChild = null;
this.rightChild = null;
}
//根据前序和中序创建二叉树
function createBinTreeByPreIn(preOrder, inOrder) {
//如果只剩一个字符,则直接创建节点并返回
if (preOrder.length === 1) {
return new BinTreeNode(preOrder.charAt(0));
}
//从前序获取当前根节点字符
var c = preOrder.charAt(0);
//中序索引和前序节点个数
var i = 0;
var number = preOrder.length;
//遍历中序序列,直到发现当前根节点
while (i < number && inOrder.charAt(i) != c) i++;
//求出左子树和右子树节点个数
var leftNumber = i;
var rightNumber = number - i - 1;
//创建当前根节点
var node = new BinTreeNode(c);
//创建左子树
if (leftNumber >= 1) {
var leftPre = preOrder.substring(1, 1 + leftNumber);
var leftIn = inOrder.substring(0, leftNumber);
node.leftChild = createBinTreeByPreIn(leftPre, leftIn);
}
//创建右子树
if (rightNumber >= 1) {
var rightPre = preOrder.substring(leftNumber + 1);
var rightIn = inOrder.substring(leftNumber + 1);
node.rightChild = createBinTreeByPreIn(rightPre, rightIn);
}
//返回当前根节点
return node;
}
//根据中序和后序创建二叉树
function createBinTreeByInPost(inOrder, postOrder) {
//如果只剩一个字符,则直接创建节点并返回
if (postOrder.length === 1) {
return new BinTreeNode(postOrder.charAt(0