前端常见算法面试题之 - 重建二叉树[JavaScript解法]

题目描述

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

实现思路

根据前序遍历和中序遍历的定义，我们不难发先输入数据可以进行如下拆分

既然我们已经分别找到了左、右子树的前序遍历和中序遍历，那么我们就可以用同样的方法分别去构建左右子树，进行递归推导

代码实现

function reConstructBinaryTree(pre, vin) {
        if(pre.length==0||vin.length==0){
                return null;
        };
        
        //前序第一个为根节点 也是中序左右子树的分割点
        var index=vin.indexOf(pre[0]);
        var left=vin.slice(0,index);//中序左子树
        var right=vin.slice(index+1);//中序右子树
        
        return {
            val:pre[0],
            //递归左右子树的前序，中序
            left:reConstructBinaryTree(pre.slice(1,index+1),left),
            right:reConstructBinaryTree(pre.slice(index+1),right)
        };
}