106. 从中序与后序遍历序列构造二叉树
难度:中等
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
思路
思路同前序与中序遍历构造二叉树差不多,只不过前序遍历取得是前序遍历数组第一个,后序遍历取得是后序遍历数组最后一个,在分割左右后序遍历数组时,后序遍历数组首先要去掉的是最后一位,先序是第一位。
即:
1.首先明确后序遍历与中序遍历的特点,后序遍历是左右根,中序遍历是左根右。
2. 因此后序遍历数组中最后一个结点一定是根节点,将这个根节点放在中序遍历中,该结点左边是左子树,结点右边是右子树。
3. 然后根据中序遍历左子树长度对应分割出后序遍历的左右子树后,再继续递归对左右子树继续重复步骤1,2,3,递归的边界条件就是左右子树为空或者是叶子结点。
代码如下
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {number[]} inorder
* @param {number[]} postorder
* @return {TreeNode}
*/
var buildTree = function(inorder, postorder) {
return diGui(postorder, inorder);
};
function diGui(postorder, inorder) {
if(!postorder.length) return null; // 空,直接返回
let value = postorder[postorder.length - 1], i; // 利用根节点在后序遍历序列中最后一位的特性
let root = new TreeNode(value);
if(postorder.length === 1) return root; // 叶子结点
// 在中序遍历序列中找到后序遍历序列中的最后一位的位置
for(i = 0; i < inorder.length; i++) {
if(value === inorder[i]) break;
}
// 根据下标,分别将中序遍历序列,和后序遍历序列给分割成左右两个子序列
let leftInorder = inorder.slice(0, i), rightInorder = inorder.slice(i + 1);
let leftpostorder = postorder.slice(0, leftInorder.length), rightpostorder = postorder.slice(leftInorder.length, postorder.length - 1);
// 让该结点左右子树继续递归
root.left = diGui(leftpostorder, leftInorder);
root.right = diGui(rightpostorder, rightInorder);
return root;
}
时间和空间复杂度如图