首发于 语雀文档@blueju

    代码:

    1. /**
    2. * Definition for a binary tree node.
    3. * class TreeNode {
    4. * val: number
    5. * left: TreeNode | null
    6. * right: TreeNode | null
    7. * constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
    8. * this.val = (val===undefined ? 0 : val)
    9. * this.left = (left===undefined ? null : left)
    10. * this.right = (right===undefined ? null : right)
    11. * }
    12. * }
    13. */
    14. function buildTree(preorder, inorder) {
    15. if (preorder.length === 0 || inorder.length === 0) return null
    16. const root = preorder[0]
    17. const rootInInorderIndex = inorder.findIndex(item => item === root)
    18. const left = inorder.slice(0, rootInInorderIndex)
    19. const right = inorder.slice(rootInInorderIndex + 1)
    20. const tree = new TreeNode(root)
    21. tree.left = buildTree(preorder.slice(1, left.length + 1), left)
    22. tree.right = buildTree(preorder.slice(left.length + 1), right)
    23. return tree
    24. };