Appearance
105. 从前序与中序遍历序列构造二叉树
题目描述
给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。
示例 1:
输入**😗* preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] 输出: [3,9,20,null,null,15,7]
示例 2:
输入: preorder = [-1], inorder = [-1] 输出: [-1]
提示:
- `1 <= preorder.length <= 3000`
- `inorder.length == preorder.length`
- `-3000 <= preorder[i], inorder[i] <= 3000`
- `preorder` 和 `inorder` 均 **无重复** 元素
- `inorder` 均出现在 `preorder`
- `preorder` **保证** 为二叉树的前序遍历序列
- `inorder` **保证** 为二叉树的中序遍历序列
难度： Medium
题解代码
javascript
/**
* 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[]} preorder
* @param {number[]} inorder
* @return {TreeNode}
*/
var buildTree = function(preorder, inorder) {
const hash = {}, end = inorder.length - 1
for (let i = 0; i < inorder.length; i++) {
hash[inorder[i]] = i
}
return genTree(preorder, 0, end, inorder, 0, end, hash)
};
function genTree (preorder, p_start, p_end, inorder, i_start, i_end, hash) {
if (p_start > p_end) return null
const root = new TreeNode(preorder[p_start])
const root_idx = hash[root.val]
const leftNum = root_idx - i_start
root.left = genTree(preorder, p_start + 1, p_start + leftNum, inorder, i_start, root_idx, hash)
root.right = genTree(preorder, p_start + 1 + leftNum, p_end, inorder, root_idx + 1, i_end, hash)
return root
}