【二叉树】根据二叉树的中序遍历和前序遍历,还原二叉树

已知二叉树的前序遍历和中序遍历:
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]);//输出后序遍历值
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值