Formatted question description: https://leetcode.ca/all/1008.html
1008. Construct Binary Search Tree from Preorder Traversal (Medium)
Return the root node of a binary search tree that matches the given preorder
traversal.
(Recall that a binary search tree is a binary tree where for every node, any descendant of node.left
has a value <
node.val
, and any descendant of node.right
has a value >
node.val
. Also recall that a preorder traversal displays the value of the node
first, then traverses node.left
, then traverses node.right
.)
Example 1:
Input: [8,5,1,7,10,12] Output: [8,5,10,1,7,null,12]![]()
Note:
1 <= preorder.length <= 100
- The values of
preorder
are distinct.
Related Topics:
Tree
Solution 1.
// OJ: https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/
// Time: O(N^2)
// Space: O(H)
class Solution {
private:
TreeNode* construct(vector<int> &preorder, int begin, int end) {
if (begin >= end) return NULL;
auto root = new TreeNode(preorder[begin]);
int i = begin + 1;
while (i < end && preorder[i] < preorder[begin]) ++i;
root->left = construct(preorder, begin + 1, i);
root->right = construct(preorder, i, end);
return root;
}
public:
TreeNode* bstFromPreorder(vector<int>& preorder) {
return construct(preorder, 0, preorder.size());
}
};
Or use find_if
:
// OJ: https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/
// Time: O(N^2)
// Space: O(H)
class Solution {
typedef vector<int>::iterator iter;
TreeNode* dfs(vector<int>& A, iter begin, iter end) {
if (begin == end) return NULL;
auto root = new TreeNode(*begin);
auto mid = find_if(begin + 1, end, [&](int x) { return x > *begin; });
root->left = dfs(A, begin + 1, mid);
root->right = dfs(A, mid, end);
return root;
}
public:
TreeNode* bstFromPreorder(vector<int>& A) {
return dfs(A, A.begin(), A.end());
}
};
Solution 2. Stack
We use a stack<pair<TreeNode*, int>> s
to keep track of the parent nodes where the first item in the pair is the parent node, and the second item is the right bound of the parent node (i.e. the grandparent’s node value).
- If the current node value is smaller than the value of the parent node at the stack top, we should add this new node as the left child of the parent node, and push
node, s.top().first->val
to the stack top. - Otherwise, we keep popping the stack top until
A[i] < s.top().second
. Then we add the new node as the right child of the parent node at the stack top, and pushnode, s.top().second
to the stack top.
// OJ: https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/
// Time: O(N)
// Space: O(H)
class Solution {
public:
TreeNode* bstFromPreorder(vector<int>& A) {
stack<pair<TreeNode*, int>> s;
s.emplace(new TreeNode(A[0]), INT_MAX);
auto root = s.top().first;
for (int i = 1; i < A.size(); ++i) {
auto node = new TreeNode(A[i]);
if (A[i] < s.top().first->val) {
s.top().first->left = node;
s.emplace(node, s.top().first->val);
} else {
while (A[i] > s.top().second) s.pop();
s.top().first->right = node;
s.emplace(node, s.top().second);
}
}
return root;
}
};
Solution 3.
// OJ: https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/
// Time: O(N)
// Space: O(H)
// Ref: https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/discuss/252232/JavaC%2B%2BPython-O(N)-Solution
class Solution {
int i = 0;
public:
TreeNode* bstFromPreorder(vector<int>& A, int bound = INT_MAX) {
if (i == A.size() || A[i] > bound) return NULL;
auto root = new TreeNode(A[i++]);
root->left = bstFromPreorder(A, root->val);
root->right = bstFromPreorder(A, bound);
return root;
}
};
Java
-
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public TreeNode bstFromPreorder(int[] preorder) { int length = preorder.length; int[] inorder = new int[length]; System.arraycopy(preorder, 0, inorder, 0, length); Arrays.sort(inorder); return buildTree(preorder, inorder); } public TreeNode buildTree(int[] preorder, int[] inorder) { if (preorder == null || preorder.length == 0) return null; TreeNode root = new TreeNode(preorder[0]); int length = preorder.length; Stack<TreeNode> stack = new Stack<TreeNode>(); stack.push(root); int inorderIndex = 0; for (int i = 1; i < length; i++) { int preorderVal = preorder[i]; TreeNode node = stack.peek(); if (node.val != inorder[inorderIndex]) { node.left = new TreeNode(preorderVal); stack.push(node.left); } else { while (!stack.isEmpty() && stack.peek().val == inorder[inorderIndex]) { node = stack.pop(); inorderIndex++; } node.right = new TreeNode(preorderVal); stack.push(node.right); } } return root; } }
-
// OJ: https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/ // Time: O(N^2) // Space: O(H) class Solution { private: TreeNode* construct(vector<int> &preorder, int begin, int end) { if (begin >= end) return NULL; auto root = new TreeNode(preorder[begin]); int i = begin + 1; while (i < end && preorder[i] < preorder[begin]) ++i; root->left = construct(preorder, begin + 1, i); root->right = construct(preorder, i, end); return root; } public: TreeNode* bstFromPreorder(vector<int>& preorder) { return construct(preorder, 0, preorder.size()); } };
-
# Definition for a binary tree node. # class TreeNode(object): # def __init__(self, x): # self.val = x # self.left = None # self.right = None class Solution(object): def bstFromPreorder(self, preorder): """ :type preorder: List[int] :rtype: TreeNode """ if not preorder: return None root = TreeNode(preorder[0]) N = len(preorder) i = 1 while i < N: if preorder[i] > preorder[0]: break i += 1 root.left = self.bstFromPreorder(preorder[1:i]) root.right = self.bstFromPreorder(preorder[i:]) return root