[LeetCode]173. Binary Search Tree Iterator

实现二叉搜索树迭代器

最新推荐文章于 2025-02-03 15:37:10 发布

原创最新推荐文章于 2025-02-03 15:37:10 发布 · 214 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 专栏收录该内容

76 篇文章

订阅专栏

本文介绍如何实现一个用于遍历二叉搜索树的迭代器，通过使用栈结构优化遍历过程，确保每次调用 next 方法都能返回 BST 中的下一个最小元素。

Problem Description

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.

思路

利用一个全局stack

Code

package q173;

import java.util.Stack;

import TreeNode.TreeNode;

public class BSTIterator {

    private Stack<TreeNode> s=new Stack<TreeNode>();
    private void PushAll(TreeNode root){
        while(root!=null){
            s.push(root);
            root=root.left;
        }
    }
    public BSTIterator(TreeNode root) {
        PushAll(root);
    }

    /** @return whether we have a next smallest number */
    public boolean hasNext() {
        return !s.isEmpty();
    }

    /** @return the next smallest number */
    public int next() {
        TreeNode tmp=s.pop();
        PushAll(tmp.right);
        return tmp.val;
    }
}