【leetcode】【173】Binary Search Tree Iterator

一、问题描述

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.

Note: next() and hasNext() should run in average O(1) time and uses O(h) memory, where h is the height of the tree.

二、问题分析

这是一道design类的题目。涉及的是二叉查找树，题目要求uses O(h) memory，因此我们不能先完全遍历完，然后再访问。也不能next()一次现遍历，结合树的非递归遍历方法，我们需要一个存储结构来暂存提前遍历的某些数据，因此可以用stack。因此结合二叉查找树的特点和树的非递归遍历，比较能想到。

三、Java AC代码

public class BSTIterator {

    private TreeNode root ;
	private LinkedList<TreeNode> stack ;
	public BSTIterator(TreeNode root) {
		this.root = root;
		stack = new LinkedList<TreeNode>();
		setNext(root);
	}
    public void setNext(TreeNode root){
        while(root!=null){
            stack.push(root);
            root = root.left;
        }
    }
	/** @return whether we have a next smallest number */
	public boolean hasNext() {
		return !stack.isEmpty();
	}

	/** @return the next smallest number */
	public int next() {
	    TreeNode node = stack.pop();
	    setNext(node.right);
		return node.val;
	}
}