leetcode题解173-二叉搜索树迭代器

问题描述

实现一个二叉搜索树迭代器。你将使用二叉搜索树的根节点初始化迭代器。

调用 next() 将返回二叉搜索树中的下一个最小的数。

示例：

BSTIterator iterator = new BSTIterator(root);
iterator.next(); //返回 3
iterator.next(); // 返回 7
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 9
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 15
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 20
iterator.hasNext(); // 返回 false

提示：
next() 和 hasNext() 操作的时间复杂度是 O(1)，并使用 O(h) 内存，其中 h 是树的高度。
你可以假设 next() 调用总是有效的，也就是说，当调用 next() 时，BST 中至少存在一个下一个最小的数。

解题思路：

要求寻找二叉搜索树的下一个最小的数，可以想到使用寻找后继的办法，但是题中要求next()和hasNext()操作时间的复杂度是O(1),因此我们可以想到使用中序遍历的方法把遍历结果集合保存起来，然后再指定一个指针，初值为-1，每次调用next()方法，指针后移一位，并将指针所指向的值返回。hasnext用来判断指针后一位所指的值是否还在集合中。而我们中序遍历的过程则可以放在构造函数的初始化过程中。

代码实现

class BSTIterator {
    List<Integer> list;         //保存遍历序列
    int pointer;                   //next指针
    public BSTIterator(TreeNode root) {
        list=new ArrayList<Integer>();
        midOrder(root);         //得到中序遍历的结果集
        pointer=-1;             //初值为-1，当时未指向任何的元素
    }
    
    /** @return the next smallest number */
    public int next() {
        pointer++;
        return list.get(pointer);
    }
    
    /** @return whether we have a next smallest number */
    public boolean hasNext() {
        int temp=pointer+1;
        if(temp>=list.size()){
            return false;
        }
        return true;
    }
    public void midOrder(TreeNode p){
        if(p!=null){
            if(p.left!=null){
                midOrder(p.left);
            }
            list.add(p.val);
            if(p.right!=null){
                midOrder(p.right);
            }
        }
    }
}