根据遍历结果构建二叉树
今天做了两道题:
105. 从前序与中序遍历序列构造二叉树
106. 从中序与后序遍历序列构造二叉树
特地来总结一下:
构造二叉树的核心思想就是利用找出根节点,将数组分为左右子树;然后对左右子树也进行这种操作划分为更小的结构,直到子树为空。
定义一个子函数用于返回该子树的根节点。
1. 前序和中序构建
根据前序遍历从根节点开始,即中左右的顺序,可以确定根节点。然后利用根节点在中序遍历中的位置(中序遍历是左中右),就可以将树分为左子树和右子树,然后对于子树们,利用同样的方法就可以在前序遍历中找出它们分别的根节点;然后重复这个操作,直到节点为空。
具体实现的时候,我这里分别记录子树在前序数组和中序数组中的起始下标和终止下标。
因为子树的长度不会改变,所以根据中序遍历算出两个子树的长度后,就可以确定子树在前序数组中的范围。
代码:
// 方法与中序后序构造的方法相同,根据前序确定根节点,利用中序划分左子树右子树
var buildTree = function(preorder, inorder) {
function apart(prestart, preend, instart, inend){
if (prestart > preend) return null;
let root = new TreeNode(preorder[prestart]