【剑指Offer】重建二叉树——JavaScript实现

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。（牛客网题目，跟原书稍有差别）

分析（适合能手画出树结构，但是不知道该从哪开始写代码的人）：前序遍历的第一个节点就是根节点，在中序中，以这个根节点（1）划分，左边就是左子树的中序遍历序列（4,7,2），然后根据该序列，从前序中刨去第一个根节点之后，取出相同长度的子序列（2,4,7），该序列就是左子树的前序遍历序列，从而我们可以这样递归得出左子树的结构。右子树同理。代码如下：

/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */
function reConstructBinaryTree(pre, vin)
{
    // write code here
    var root = new TreeNode(pre[0]);
    if(pre.length === 1)return root;//如果只有一个节点
    var index = vin.indexOf(pre[0]);
    var leftLen = index;
    var leftPre = pre.slice(1,leftLen + 1);//取到左子树的前序遍历序列
    var rightPre = pre.slice(leftLen + 1);//右子树的前序遍历序列
    var leftVin = vin.slice(0,index);//左子树的中序遍历序列
    var rightVin = vin.slice(index + 1);//右子树的中序。。。。
    if(leftLen > 0){//如果左边有节点的话
        root.left = reConstructBinaryTree(leftPre,leftVin);
    }
    if(rightVin.length > 0){//如果右边有节点的话
       	root.right = reConstructBinaryTree(rightPre,rightVin); 
    }
    return root;
}

上面代码可能不够严谨，对一些边界条件处理不够，详细请看原书。