剑指offer(4)重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

解题代码:

/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */
function reConstructBinaryTree(pre, vin)
{
    // write code here
    if( pre.length == 0 || vin.length == 0){
        return null;
    }
    //前序遍历顺序:根左右;中序遍历顺序:左根右
    //前序遍历的第一个结点为树的根节点,找出其在中序遍历中的位置,左边的为左子树结点,右边的为右子树结点
    var idx = vin.indexOf(pre[0]);
    var vinLeft = vin.slice(0,idx);
    var vinRight = vin.slice(idx+1);
    var preLeft = pre.slice(1,idx+1);
    var preRight = pre.slice(idx+1);
    var head = new TreeNode(pre[0]);
    //递归左右子树的前序遍历序列,中序遍历序列
    head.left = reConstructBinaryTree(preLeft,vinLeft);
    head.right = reConstructBinaryTree(preRight,vinRight);
    return head;
}

猜你喜欢

转载自www.cnblogs.com/3yleaves/p/9588787.html