剑指offer(4)重建二叉树
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题代码:
/* 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;
}
//前序遍历顺序:根左右;中序遍历顺序:左根右
//前序遍历的第一个结点为树的根节点,找出其在中序遍历中的位置,左边的为左子树结点,右边的为右子树结点
var idx = vin.indexOf(pre[0]);
var vinLeft = vin.slice(0,idx);
var vinRight = vin.slice(idx+1);
var preLeft = pre.slice(1,idx+1);
var preRight = pre.slice(idx+1);
var head = new TreeNode(pre[0]);
//递归左右子树的前序遍历序列,中序遍历序列
head.left = reConstructBinaryTree(preLeft,vinLeft);
head.right = reConstructBinaryTree(preRight,vinRight);
return head;
}