代码如下
function reConstructBinaryTree(pre, vin)
{//pre是先序遍历数组,vin是中序遍历数组
if(pre.length === 0 || vin.length === 0){
return null;//如果任意数组为空就不能重建了
}
var root = pre[0]; //创建根节点,根节点是前序遍历的第一个数
var tree = new TreeNode(root);//创建重建树的新节点
var index = vin.indexOf(root);//找到中序遍历根节点所在位置
tree.left = reConstructBinaryTree(pre.slice(1,index+1),vin.slice(0,index));
tree.right = reConstructBinaryTree(pre.slice(index+1),vin.slice(index+1));
//对于中序遍历,根节点左边的节点即左子树,根节点右边的节点即右子树,对于先序遍历,除去第一个为根节点外,有index个节点的左子树,其余为右子树
return tree;
}
附加:中序遍历+后序遍历重构二叉树代码
基本和前一个差不多!理解一个就都懂了
function TreeNode(x) {
this.val = x;
this.left = null;
this.right = null;
}
function reConstructBinaryTree(pos, vin)
{
if(!pos || vin.length === 0){
return null;
}
//创建根节点,根节点是后序遍历的最后一个数
var root = pos[pos.length-1];
var tree = new TreeNode(root);
//找到中序遍历根节点所在位置
var index = vin.indexOf(root);
//对于中序遍历,根节点左边的节点即左子树,根节点右边的节点即右子树
tree.left=reConstructBinaryTree(pos.slice(0,index),vin.slice(0,index))
tree.right=reConstructBinaryTree(pos.slice(index,pos.length-1),vin.slice(index+1))
return tree;
}