从前序与中序遍历序列构造二叉树(JS实现)
从前序与中序遍历序列构造二叉树(JS实现)
1 题目
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/
9 20
/
15 7
2 思路
这道题的主要思路是使用递归的方法来构造整个树,首先前序遍历中,*个节点是根节点,我们就可以将中序遍历数组划分为两个子数组,分别对应根节点的左右子树。然后针对左子树,前序遍历的第二个节点即为左子树的根节点,我们又可以将左子树的中序遍历数组进一步划分为两个子数组,就这样每次取前序遍历的一个元素,将中序数组层层划分下去
3代码
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {number[]} preorder
* @param {number[]} inorder
* @return {TreeNode}
*/
var buildTree = function(preorder, inorder) {
function d(preArr, inArr, begin, end) {
if (begin > end) return null;
let index = inArr.indexOf(preArr[0]); //找到当前子树的根节点
let node = new TreeNode(preArr[0]); //新建子树的根节点
preArr.shift();
node.left = d(preArr, inArr, begin, index – 1); //继续划分左子树
node.right = d(preArr, inArr, index + 1, end); //继续划分右子树
return node;
}
return d(preorder, inorder, 0, inorder.length – 1);
function TreeNode(val) {
this.val = val;
this.left = this.right = null;
}
};