【LeetCode-173】Binary Search Tree Iterator

cold2wind

于 2016-03-18 14:58:23 发布

阅读量257

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： LeetCode-173 Binary Search Tree I java

本文链接：https://blog.youkuaiyun.com/cold2wind/article/details/50923528

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();
	}
}