题目链接:https://leetcode.com/problems/binary-search-tree-iterator/
实现一个二叉搜索树迭代器。你将使用二叉搜索树的根节点初始化迭代器。
调用 next()
将返回二叉搜索树中的下一个最小的数。
示例:
BSTIterator iterator = new BSTIterator(root); iterator.next(); // 返回 3 iterator.next(); // 返回 7 iterator.hasNext(); // 返回 true iterator.next(); // 返回 9 iterator.hasNext(); // 返回 true iterator.next(); // 返回 15 iterator.hasNext(); // 返回 true iterator.next(); // 返回 20 iterator.hasNext(); // 返回 false
提示:
next()
和hasNext()
操作的时间复杂度是 O(1),并使用 O(h) 内存,其中 h 是树的高度。- 你可以假设
next()
调用总是有效的,也就是说,当调用next()
时,BST 中至少存在一个下一个最小的数。
思路:
在时间复杂度和空间复杂度的限制下,我只好去拜读大佬们的做法。
看到一个
O(h) memory,
hasNext() O(1) time
next() O(h) time.的做法。
大佬的代码思路:
由于内存的限制,而且每次next返回的是最小值,所以直接用Stack保存树的左侧节点,刚好是O(h) memory;
然后只要stack不为空,hasNext()就为true,也满足。
美中不足的是next并不是O(1)时间,而是O(h),
每次next拿到节点时就会把这个节点的右子树(仍是左链)入栈。这样保证了每次得到的next都是最小值。
AC 58ms:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class BSTIterator {
Stack<TreeNode> sk=new Stack();
public BSTIterator(TreeNode root) {
pushLeft(root);
}
/** @return the next smallest number */
public int next() {
TreeNode temp=sk.pop();
pushLeft(temp.right);
return temp.val;
}
/** @return whether we have a next smallest number */
public boolean hasNext() {
return !sk.isEmpty();
}
public void pushLeft(TreeNode root){
while(root!=null){
sk.push(root);
root=root.left;
}
}
}
/**
* 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();
*/