Solve LeetCode 105: Construct Binary Tree from Preorder and Inorder Traversal in Python with a indexed recursive partition approach. The key is to make the state invariant explicit, so the implementation and complexity follow naturally.
This guide paraphrases the task and does not reproduce LeetCode’s prompt. Use the official page for the complete statement, examples, constraints, and submission runner.
| Difficulty | Medium |
|---|---|
| Topic | Binary Tree General |
| Reusable pattern | indexed recursive partition |
| Complexity | O(n) time and O(n) space |
What the problem is testing
Take roots from preorder in order and use an inorder index map to split each subtree into left and right ranges.
Algorithm
- Take roots from preorder in order and use an inorder index map to split each subtree into left and right ranges.
- Maintain this invariant: The preorder cursor always identifies the root of the current inorder range.
- Continue until every input item or reachable state has been resolved, then return the accumulated result.
Python solution
LeetCode provides the list, tree, or graph node definition used by the method.
from collections import Counter, defaultdict, deque, OrderedDict
import random
class Solution:
def buildTree(self, preorder, inorder):
position = {value: i for i, value in enumerate(inorder)}
pre_index = 0
def build(left, right):
nonlocal pre_index
if left > right:
return None
value = preorder[pre_index]
pre_index += 1
root = TreeNode(value)
split = position[value]
root.left = build(left, split - 1)
root.right = build(split + 1, right)
return root
return build(0, len(inorder) - 1)Why this is correct
The proof follows the maintained state: The preorder cursor always identifies the root of the current inorder range. Each iteration preserves that claim while permanently resolving at least one position, node, interval, or search state. When the loop or recursion ends, every candidate required by the problem has therefore been included or ruled out, so the returned value is correct.
Complexity
O(n) time and O(n) space. The stated auxiliary space excludes the returned output unless the output is the data structure being built.
Edge cases
Values are unique; empty inorder ranges return no node.
Tested reference code
This implementation is included in the site’s downloadable 100-solution Python library. The complete suite compiles every solution and runs a behavioral assertion for every problem before publication.
Browse the searchable 100 LeetCode Python Solutions hub. Previous: 101. Symmetric Tree · Next: 106. Construct Binary Tree from Inorder and Postorder Traversal