173. Binary Search Tree Iterator

最新推荐文章于 2024-05-04 15:02:13 发布

原创最新推荐文章于 2024-05-04 15:02:13 发布 · 159 阅读

0 ·

CC 4.0 BY-SA版权

Tree 专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一种使用队列实现二叉搜索树迭代器的方法。通过深度优先遍历将节点依次加入队列，实现了返回下一个最小数值的功能，并判断是否还有下一个最小数。

题目描述

在这里插入图片描述

题目链接

https://leetcode.com/problems/binary-search-tree-iterator/

方法思路

Approach:using Queue

class BSTIterator {
    //Runtime: 59 ms, faster than 91.88% 
    //Memory Usage: 52.7 MB, less than 9.65% 
    TreeNode node;
    Queue<TreeNode> queue;
    public BSTIterator(TreeNode root) {
        node = root;
        queue = new LinkedList<>();
        dfs(root);
    }
    
    /** @return the next smallest number */
    public int next() {
        return queue.poll().val;
    }
    
    /** @return whether we have a next smallest number */
    public boolean hasNext() {
        if(queue.peek() != null)
            return true;
        return false;
    }
    
    public void dfs(TreeNode root){
        if(root == null) return;
        dfs(root.left);
        queue.offer(root);
        dfs(root.right);
    }
}