박종훈 알고리즘 블로그

(Leetcode) 105 - Construct Binary Tree from Preorder and Inorder Traversal

기회가 되어 달레님의 스터디에 참여하여 시간이 될 때마다 한문제씩 풀어보고 있다.

https://neetcode.io/practice


https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

내가 작성한 풀이

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        Map<Integer, Integer> inorderIndexMap = new HashMap<>();
        for (int i = 0; i < inorder.length; i++) {
            inorderIndexMap.put(inorder[i], i);
        }

        return buildTree(inorderIndexMap, new Traversal(preorder), new Traversal(inorder));
    }

    public TreeNode buildTree(Map<Integer, Integer> inorderIndexMap, Traversal preorderTraversal, Traversal inorderTraversal) {
        if(preorderTraversal.start > preorderTraversal.end) {
            return null;
        }

        TreeNode treeNode = new TreeNode(preorderTraversal.getFirst());
        if(preorderTraversal.start == preorderTraversal.end) {
            return treeNode;
        }

        int rootIndex = inorderIndexMap.get(preorderTraversal.getFirst());
        int leftSize = rootIndex - inorderTraversal.start;
        treeNode.left = buildTree(
                inorderIndexMap,
                preorderTraversal.subIterator(preorderTraversal.start + 1, preorderTraversal.start + leftSize),
                inorderTraversal.subIterator(inorderTraversal.start, rootIndex - 1)
        );
        treeNode.right = buildTree(
                inorderIndexMap,
                preorderTraversal.subIterator(preorderTraversal.start + leftSize + 1, preorderTraversal.end),
                inorderTraversal.subIterator(rootIndex + 1, inorderTraversal.end)
        );

        return treeNode;
    }
}

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode() {
    }

    TreeNode(int val) {
        this.val = val;
    }

    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }

    @Override
    public String toString() {
        return "{" +
                "val=" + val +
                ", left=" + left +
                ", right=" + right +
                '}';
    }
}

class Traversal {
    int[] array;
    int start;
    int end;

    public Traversal(int[] array) {
        this.array = array;
        this.start = 0;
        this.end = array.length - 1;
    }

    private Traversal(int[] array, int start, int end) {
        this.array = array;
        this.start = start;
        this.end = end;
    }

    public Traversal subIterator(int start, int end) {
        return new Traversal(array, start, end);
    }

    public int getFirst() {
        return array[start];
    }

    public Traversal(int start, int end) {
        this.start = start;
        this.end = end;
    }

    @Override
    public String toString() {
        return "{" +
                "start=" + start +
                ", end=" + end +
                '}';
    }
}

TC, SC

시간 복잡도는 O(n)이고, 공간 복잡도는 O(n)이다.