Skip to content

Code:

js
import {
    TreeNode,
    ArrayToTree
} from '../common.js'

/**
 * 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) {
    let preorderIndex = 0
    const map = new Map()

    for (let i = 0; i < inorder.length; i++) {
        map.set(inorder[i], i)        
    }

    const dfs = (i, j) => {
        if (i > j) {
            return null
        }

        const val = preorder[preorderIndex++]
        const middle = map.get(val)
        
        return new TreeNode(val, dfs(i, middle - 1), dfs(middle + 1, j))
    }


    return dfs(0, preorder.length - 1)
};

console.log(buildTree([1, 2, 3], [3, 2, 1]))

❤ With Algorithm