题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
我们知道前序遍历序列第一个值就是根节点,而在中序序列中,此根节点左边的序列都属于左子树,右边的值都属于右子树,例如:
前序:{1,2,4,7,3,5,6,8}里1是根节点,那么在中序里:{4,7,2,1,5,3,8,6},1之前的{4,7,2}都属于左子树,{5,3,8,6}都属于右子树。此时我们把这个二叉树的左右子树再看成一个整体,先看左子树的中序序列{4,7,2},在前序里是{2,4,7},2是开头,那么2就是左子树的根节点,从中序{4,7,2}可知2这个节点只有左子树,那么再继续往下细分(就用到函数递归)。。。右子树同理
代码:
/* function TreeNode(x) {
this.val = x;
this.left = null;
this.right = null;
} */
function reConstructBinaryTree(pre, vin)
{
// write code here
var result = null;
if (pre.length === 1){
result = {
val: pre[0],
left: null,
right: null
}
} else 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)
}
}
return result;
}