剑指offer-重建二叉树-JavaScript

题目

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

思路

前序遍历：若二叉树为空，则空操作；否则：1.访问根节点；2.先序遍历左字树；3.先序遍历右子树；

中序遍历：若二叉树为空，则空操作；否则：1.中序遍历左子树；2.访问根节点；3.中序遍历右子树；

因为是树的结构，一般都是用递归来实现。

根据前序遍历的性质，第一个元素就是root，那么下面的工作就是如何确定root的左右子树的范围。

根据中序遍历的性质，root元素前面都是root的左子树，后面都是root的右子树。那么我们只要找到中序遍历中root的位置，就可以确定左右子树的范围。

正如上面所说，只需要将确定的左右子树安到root上即可。递归要注意出口，假设最后只有一个元素了，那么就要返回。

本题重建后的二叉树：

JavaScript代码

/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */
function reConstructBinaryTree(pre, vin)
{
    var result;
    if (pre.length > 1) {
        var root = pre[0];
        var vinRootIndex = vin.indexOf(root);
        var vinLeft = vin.slice(0, vinRootIndex);
        var vinRight = vin.slice(vinRootIndex + 1, vin.length);
        pre.shift();
        var preLeft = pre.slice(0, vinLeft.length);
        var preRight = pre.slice(vinLeft.length, pre.length);
        result = {
            val: root,
            left: reConstructBinaryTree(preLeft, vinLeft),
            right: reConstructBinaryTree(preRight, vinRight)
        }
    } else if (pre.length === 1) {
        result = {
            val: pre[0],
            left: null,
            right: null
        }
    }
    return result;
}