Skip to content

leet code

943 - Construct Binary Tree from Preorder and Inorder Traversal

저번 문제는 inorder와 postorder를 이용해서 원래 tree를 알아내는 것이었습니다.

이번에는 postorder 대신 preorder traversal이 있습니다.

  • inorder
  • left -> root -> right
  • preorder
  • root->left->right

이번에도 같은 전략으로 inorder를 이용해서 left와 right를 자르고 붙이는 전략이 좋겠습니다.

함수 정의

1
2
TreeNode* solve(TreeNode* root, int inorder_start, int inorder_end, int preorder_start, int preorder_end);
// root의 left node와 right node를 연결해서 내 보냄

답을 푸는 방법은 확실하게 알았지만, 좀 더 효율적으로 구현하는 것을 연습해야겠습니다.

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        this->preorder = preorder;
        this->inorder = inorder;

        return solve(0, inorder.size(), 0, preorder.size());
    }

    TreeNode* solve(int inorderStart, int inorderEnd, int preorderStart, int preorderEnd)
    {
        if(preorderEnd - preorderStart == 1)
        {
            return new TreeNode(this->preorder[preorderStart]);
        }
        if(preorderEnd - preorderStart <= 0)
        {
            return nullptr;
        }
        // 1. get root value
        int rootVal = this->preorder[preorderStart];

        // 2. find index from root in inorder
        int rootIndex;
        for(int i=inorderStart; i<inorderEnd; ++i)
        {
            if(this->inorder[i]==rootVal)
            {
                rootIndex = i - inorderStart;
                break;
            }
        }

        TreeNode* root = new TreeNode(rootVal);

        TreeNode* left = solve(inorderStart, inorderStart+rootIndex, preorderStart+1, preorderStart + rootIndex+1);
        if(left)
        {
            root->left = left;
        }

        TreeNode* right = solve(inorderStart+rootIndex+1, inorderEnd, preorderStart + rootIndex+1, preorderEnd);
        if(right)
        {
            root->right = right;
        }

        return root;

    }

private:
    vector<int> inorder;
    vector<int> preorder;
};