105. [✔][M]从前序与中序遍历序列构造二叉树
给定两个整数数组 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
保证 为二叉树的中序遍历序列
题解
// @ts-ignore
function buildTree(preorder: number[], inorder: number[]): TreeNode | null {
let inorderMap = {}
for (let i = 0; i < inorder.length; i++) {
inorderMap[inorder[i]] = i;
}
// @ts-ignore
const build = (preorder: number[], preStart: number, preEnd: number, inorder: number[], inStart: number, inEnd: number,) => {
if (preStart > preEnd) {
return null;
}
let rootVal = preorder[preStart];
let index = inorderMap[rootVal];
let leftSize = index - inStart;
// @ts-ignore
let root = new TreeNode(rootVal);
root.left = build(preorder, preStart + 1, preStart + leftSize, inorder, inStart, index - 1);
root.right = build(preorder, preStart + leftSize + 1, preEnd, inorder, index + 1, inEnd);
return root;
}
return build(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1);
};
// @lc code=end