已知二叉树的前序遍历和中序遍历:
PreOrder: 1 2 4 3 5 6
InOrder: 4 2 1 5 3 6
如何还原这颗二叉树,并求出他的后序遍历?
求解:
基于一个事实:
中序遍历一定是 { 左子树中的节点集合 },root,{ 右子树中的节点集合 };
而前序遍历的作用就是找到每颗子树的root位置。
// 节点Node类
function Node(data, left, right) {
this.data = data;
this.left = left;
this.right = right;
}
/* 根据中序遍历和前序遍历创建二叉树
* param:
* inorder——中序遍历数组
* preorder——前序遍历数组
* length——数组长度
* return:
* node——构建好的二叉树的根
*/
function BTformOrderings(inorder, preorder, length) {
if (length != 0){
var rootIndex = 0;
//寻找中序遍历中通过前序遍历获得的根的位置
for (; rootIndex < length; rootIndex++) {
if (inorder[rootIndex] == preorder[0]) {
break;
}
}
//通过根节点的位置将中序遍历数组切割
var leftInorder = inorder.slice(0, rootIndex);
var rightInorder = inorder.slice(rootIndex + 1);
//通过递归得到左右子树
var leftNode = BTformOrderings(leftInorder, preorder.slice(1, leftInorder.length + 1), leftInorder.length);
var rightNode = BTformOrderings(rightInorder, preorder.slice(leftInorder.length + 1), rightInorder.length);
//创建二叉树根节点
var n = new Node(preorder[0],leftNode,rightNode);
//console.log(n);
return n;
}
}
//后序遍历
function postorder(node){
if(node != null){
postorder(node.left);
postorder(node.right);
console.log(node.data);
}
}
var preorder = [1, 2, 4, 3, 5, 6];
var inorder = [4, 2, 1, 5, 3, 6];
var n = BTformOrderings( inorder,preorder, preorder.length);
postorder(n);
改进:
上面的代码写得不够简洁。题目只要求输出后续遍历,并没有要求建树。所以,不需要去计算出左子树与右子树,也不需要去return node。
function BTformOrderings(inorder, preorder, length) {
if (length != 0){
var rootIndex = 0;
for (; rootIndex < length; rootIndex++) {
if (inorder[rootIndex] == preorder[0]) {
break;
}
}
var leftInorder = inorder.slice(0, rootIndex);
var rightInorder = inorder.slice(rootIndex + 1);
BTformOrderings(leftInorder, preorder.slice(1, leftInorder.length + 1), leftInorder.length);
BTformOrderings(rightInorder, preorder.slice(leftInorder.length + 1), rightInorder.length);
console.log(preorder[0]);//输出后序遍历值
}
}