【LeetCode-173】Binary Search Tree Iterator

最新推荐文章于 2023-02-25 23:52:55 发布

原创最新推荐文章于 2023-02-25 23:52:55 发布 · 258 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode-173 #Binary Search Tree I #java

leetcode 专栏收录该内容

80 篇文章

订阅专栏

本文介绍了一种使用栈实现二叉树中序遍历迭代器的方法，通过从右子节点开始遍历并压入栈中，最终按从大到小的顺序输出节点值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这道题的关键在于在建立迭代器的时候，将二叉树遍历一遍，按照从大到小的顺序压入栈中，具体代码如下

public class BSTIterator {

	public class TreeNode {
		int val;
		TreeNode left;
		TreeNode right;

		TreeNode(int x) {
			val = x;
		}
	}

	Stack<Integer> data = new Stack<Integer>();
	public BSTIterator(TreeNode root) {
		helpFunction(root);
	}
	
	
	private void helpFunction(TreeNode root){
		if(root == null){
			return;
		}
		
		if(root.right != null){
			helpFunction(root.right);
		}
		
		data.push(root.val);
		
		if(root.left != null){
			helpFunction(root.left);
		}
	}
	
	
	/** @return whether we have a next smallest number */
	public boolean hasNext() {
		return data.size() != 0;
	}

	/** @return the next smallest number */
	public int next() {
		return data.pop();
	}
}