[ PS ] : Construct Binary Tree from Preorder and Inorder Traversal

Author: uraflower

construct-binary-tree-from-preorder-and-inorder-traversal/uraflower.js

@@ -0,0 +1,49 @@
+// 접근법
+// preorder: v l r
+// inorder: l v r
+
+// preorder의 가장 첫 요소는 무조건 root
+// inorder에서 root보다 왼쪽에 있는 요소는 전부 left임
+// 여기서 중복되는 value가 없어야 하는데 문제에서 없음을 보장함
+// 따라서 preorder[0]을 inorder에서 찾고
+// 그 왼쪽, 오른쪽으로 배열을 나눠서 이걸 반복
+
+/**
+ * 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}
+ */
+const buildTree = function (preorder, inorder) {
+    let preorderIndex = 0;
+    const map = inorder.reduce((map, val, index) => {
+        map[val] = index;
+        return map;
+    }, {});
+
+    function build(start, end) {
+        if (start > end) {
+            return null;
+        }
+
+        const root = new TreeNode(preorder[preorderIndex++]);
+        const index = map[root.val];
+
+        root.left = build(start, index - 1);
+        root.right = build(index + 1, end);
+
+        return root;
+    }
+
+    return build(0, inorder.length - 1);
+};
+
+// 시간복잡도: O(n)
+// 공간복잡도: O(n)