LeetCode(105):重建二叉树
题目描述
![image-20201118093922604](https://i.loli.net/2020/11/18/QMWU2418DImaEep.png)
解题思路:递归
前序遍历:根节点->左子节点->右子节点
中序遍历:左子节点->根节点->右子节点
针对前序和中序遍历的特点,我们不难得出以下思路
在每一轮递归中:
1、用preorder的头部值 初始化当前的root节点
2、在传入的2个数组的基础上,划分出当前root的左子树的前序/中序遍历数组、以及右子树的前序/中序遍历数组
3、调用自身,将左子树和右子树挂载到当前root上
4、返回当前root
下面我们具体来看第2个步骤如何实现
我们可以利用js提供的 indexOf 数组方法,找出当前root在inorder数组中的位置(记为 rootIndex)
因此,在inorder数组中,左子树和右子树的inorder数组分别位于 rootIndex 的左边和右边
这样一来,我们就可以利用 js 提供的 slice 方法,在原数组的基础上进行切割,从而得到左右子树的前序/中序遍历数组
除此之外,我们还要考虑一些特殊的情况:
- 传入的前序/中序遍历数组为空:返回空
- 传入的前序/中序遍历数组长度为1:当前为叶子节点
- rootIndex 等于 0 :当前root节点没有左子树
- rootIndex + 1 等于 数组的长度 :当前 root节点没有右子树
代码实现(Javascript)
/**
* 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) {
if(preorder.length===0 || inorder.length===0){
return null
}
if(preorder.length === 1 && inorder.length === 1){//没有左右子树
return new TreeNode(preorder[0])
}
var root=new TreeNode(preorder[0])
let rootIndex=inorder.indexOf(root.val)
if(rootIndex!==0){ // 有左子树
var leftinorder=inorder.slice(0,rootIndex)
if(rootIndex + 1 >= preorder.length){//有左子树,没有右子树
var leftpreorder=preorder.slice(1)
root.left = buildTree(leftpreorder, leftinorder)
root.right= buildTree([],[])
}else{ // 有左右子树
var leftpreorder=preorder.slice(1,rootIndex+1)
var rightinorder=inorder.slice(rootIndex+1)
var rightpreorder=preorder.slice(rootIndex+1)
root.left = buildTree(leftpreorder, leftinorder)
root.right = buildTree(rightpreorder, rightinorder)
}
}else{ // 没有左子树
var rightinorder=inorder.slice(rootIndex+1)
var rightpreorder=preorder.slice(rootIndex+1)
root.left = buildTree([],[])
root.right = buildTree(rightpreorder, rightinorder)
}
return root
};