跳到主要内容

105. [✔][M]从前序与中序遍历序列构造二叉树

给定两个整数数组 preorderinorder ,其中 preorder 是二叉树的先序遍历inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。

示例 1:

img

输入: 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
  • preorderinorder无重复 元素
  • 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