05-重建二叉树

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

首先，需要熟悉二叉树三种遍历方式的遍历过程：

• 前序遍历：
  1.访问根节点
  2.前序遍历左子树
  3.前序遍历右子树
• 中序遍历：
  1.中序遍历左子树
  2.访问根节点
  3.中序遍历右子树
• 后序遍历：
  1.后序遍历左子树
  2.后序遍历右子树
  3.访问根节点

注意点： 有关树、链表的题一般情况下会用到递归的思想， 尤其是树的遍历。

所以， 已知前序遍历和中序遍历，构建二叉树的步骤是这样的：

1. 确定根节点， 获取左子树，获取右子树
2. 在左子树中递归
3. 在右子树中递归
4. 得到根节点。

代码如下：

/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */
function reConstructBinaryTree(pre, vin)
{
    // write code here
    // 首先判断输入的这个前序或中序序列是不是为空
    if(pre.length == 0 || vin.length == 0) return null;
    
    // 首先要明确前序序列与中序序列之间的关系：  
    // 前序的第一个元素是二叉树的根节点， 而在中序序列中这个元素的左侧为这个根节点的左子树，右侧为这个根节点的右子树
    // 使用递归的思想，直到左右子树全部遍历完成，
    
    // 注意: 该题目要求是返回构建出来的这个二叉树！！！
    const index = vin.indexOf(pre[0]);  // 获取根节点在中序序列中的索引位置
    const left = vin.slice(0,index);  // 获取这个根节点的左子树
    const right = vin.slice(index + 1);  // 获取这个根节点的右子树
    
    return {
        val: pre[0],  // 根节点
        left: reConstructBinaryTree(pre.slice(1, index+1), left),  // 递归， 这时的pre是左子树的前序， vin是左子树的中序
        right: reConstructBinaryTree(pre.slice(index+1), right)  // 递归， 这时的输入参数pre是右子树的前序，Vin是右子树的中序
    }
}

如果该题目的要求是：已知前序序列和中序序列返回后序遍历结果呢？

/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */

function reConstructBinaryTree(pre, vin) {
    // write code here
    // 首先判断输入的这个前序或中序序列是不是为空
    if (pre.length == 0 || vin.length == 0) return null;

    // 首先要明确前序序列与中序序列之间的关系：  
    // 前序的第一个元素是二叉树的根节点， 而在中序序列中这个元素的左侧为这个根节点的左子树，右侧为这个根节点的右子树
    // 使用递归的思想，直到左右子树全部遍历完成，

    // 注意: 该题目要求是返回构建出来的这个二叉树！！！
    const index = vin.indexOf(pre[0]); // 获取根节点在中序序列中的索引位置
    const left = vin.slice(0, index); // 获取这个根节点的左子树
    const right = vin.slice(index + 1); // 获取这个根节点的右子树

    return {
        val: pre[0], // 根节点
        left: reConstructBinaryTree(pre.slice(1, index + 1), left), // 递归， 这时的pre是左子树的前序， vin是左子树的中序
        right: reConstructBinaryTree(pre.slice(index + 1), right) // 递归， 这时的输入参数pre是右子树的前序，Vin是右子树的中序
    };
}

// 递归函数
function postOrderTraversalNode(node, handler) {

    if (node != null) {
        // 后序左遍历
        postOrderTraversalNode(node.left, handler);

        // 后序右遍历
        postOrderTraversalNode(node.right, handler);

        handler(node.val);
    }
}


// 根据已有的二叉树来实现后序遍历
function postOrderTraversal(pre, vin, handler) {
    // 获取二叉树
    var bst = reConstructBinaryTree(pre, vin);

    // 递归遍历
    postOrderTraversalNode(bst, handler);

}


// 测试
var pre = [1, 2, 4, 7, 3, 5, 6, 8];
var vin = [4, 7, 2, 1, 5, 3, 8, 6];

var post = '';
postOrderTraversal(pre, vin, function (key) {  
    post += key + ' ';
});

console.log(post);  // 7 4 2 5 8 6 3 1