Implement an iterator over a binary search tree (BST). Your iterator will be initialized with the root node of a BST.
Calling next()
will return the next smallest number in the BST.
Note: next()
and hasNext()
should run in average O(1) time and uses O(h) memory, whereh is the height of the tree.
Credits:
Special thanks to @ts for adding this problem and creating all test cases.
Have you met this question in a real interview?
思路:next()
and hasNext()
should run in average O(1) time and uses O(h) memory
实际上这题就跟inorder travel stack的写法一模一样,就是每次只用返回stack最顶端的元素,如果pop出来的node还有右子树的话,那么就把右边的子数的左边全部push进去,然后再进行循环。
/**
* Definition for a binary tree node.
* public 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;
* }
* }
*/
class BSTIterator {
Stack<TreeNode> stack;
public BSTIterator(TreeNode root) {
this.stack = new Stack<>();
pushLeft(root, stack);
}
private void pushLeft(TreeNode root, Stack<TreeNode> stack) {
while(root != null) {
stack.push(root);
root = root.left;
}
}
public int next() {
TreeNode node = stack.pop();
if(node.right != null) {
pushLeft(node.right, stack);
}
return node.val;
}
public boolean hasNext() {
return !stack.isEmpty();
}
}
/**
* Your BSTIterator object will be instantiated and called as such:
* BSTIterator obj = new BSTIterator(root);
* int param_1 = obj.next();
* boolean param_2 = obj.hasNext();
*/