重建二叉树

重建二叉树

题目描述

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

思路：

1. 前序遍历数组的第一个结点就是二叉树的根结点p
2. 在中序遍历中找到二叉树的根结点所在的位置index，在根结点左右的子数组分别是二叉树的左右子树;
3. 前序遍历的第二个位置到p，也是左子树的前序遍历子数组，剩下的p右边的就是右子树的前序遍历子数组
4. 递归生成左右子树，得到二叉树

/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */
function reConstructBinaryTree(pre, vin)
{
    if(!pre||pre.length===0) return;
    // 获取根结点
    var root = new TreeNode(pre[0]);
    //查找根结点在中序遍历序列中的位置
    var index = vin.indexOf(root.val);
    //递归生成左右子树
    root.left = reConstructBinaryTree(pre.slice(1,index+1),vin.slice(0,index));
    root.right = reConstructBinaryTree(pre.slice(index+1),vin.slice(index+1));
    return root;
}