题解 | #重建二叉树#

/*
 * function TreeNode(x) {
 *   this.val = x;
 *   this.left = null;
 *   this.right = null;
 * }
 */
/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param preOrder int整型一维数组 
 * @param vinOrder int整型一维数组 
 * @return TreeNode类
 */
function reConstructBinaryTree( preOrder ,  vinOrder ) {
    // write code here
    // 这一道题的关键是要知道，preOrder的第一个就是最大的根节点，
    // 以它为基础，我们可以将vinOrder划分为两个部分，处于左子树的节点和处于右子树的节点
    // 然后对于两种节点，分别递归，返回各自的根节点 

    if(preOrder.length ==0){return null}
    let div = vinOrder.indexOf(preOrder[0])
    let leftI = vinOrder.slice(0,div) //左子树的中序
    let rightI = vinOrder.slice(div+1)//右子树的中序
    
    let leftP = preOrder.slice(1,1+leftI.length)//左子树的前序
    let rightP = preOrder.slice(1+leftI.length)//右子树的前序

    // 接下来获取左右子树的根节点

    let left = reConstructBinaryTree( leftP ,  leftI )
    let right = reConstructBinaryTree( rightP ,  rightI )

    let root = new TreeNode(preOrder[0]);
    root.left =left
    root.right =right
    return root
}
module.exports = {
    reConstructBinaryTree : reConstructBinaryTree
};