首页 > 代码库 > 《剑指offer》— JavaScript(4)重建二叉树

《剑指offer》— JavaScript(4)重建二叉树

重建二叉树

题目描述

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


实现代码

function reConstructBinaryTree(pre, vin)
{
    if(!pre || pre.length===0){
        return;
    }
    
    var root={
        val:pre[0]
    };
    
    for(var i=0;i<pre.length;i++){
        if(vin[i]==pre[0]){
            root.left= reConstructBinaryTree(pre.slice(1,i+1),vin.slice(0,i));
            root.right= reConstructBinaryTree(pre.slice(i+1),vin.slice(i+1));
        }
    }
    
    return root;
}

相关知识

  二叉树是每个节点最多有两个子树的树结构。
  前序遍历:首先访问根,再先序遍历左子树,最后先序遍历右子树。
  中序遍历:首先中序遍历左子树,再访问根,最后中序遍历右子树。
  后序遍历:首先后序遍历左子树,再后序遍历右子树,最后访问根。

思路

  先序遍历第一个位置是根节点treeNode,找到中序遍历的根节点位置在中间i;

       那么中序遍历i的左侧就是左字数的中序数组,右侧是右子树的中序数组

       先序遍历的第二个起的

《剑指offer》— JavaScript(4)重建二叉树