剑指Offer专项突破版（55）—— 二叉搜索树迭代器

题目

剑指 Offer II 055. 二叉搜索树迭代器

思路

解决这个问题有很多种不同的方法：

允许修改二叉树

如果允许修改输入的二叉搜索树，则可以在初始化迭代器时将它展平成除叶节点之外其他节点只有一个右子节点。这时二叉搜索树看起来像一个链表。然后用一个指针P指向展平后的二叉搜索树的根节点，如果指针P指向的节点不为null，那么函数hasNext将返回true。每次函数next被调用时都返回指针P指向节点的值，并将指针P朝着指向右子节点的指针向前移动一次

• 空间复杂度：如果按照中序遍历的顺序将二叉搜索树展平，那么展平的过程中需要一个大小为O（h）的栈

• 缺点：需要修改原有结构

预先将节点放到链表中

如果不允许修改输入的二叉搜索树，那么可以在迭代器初始化时另外创建一个链表保存二叉树所有节点的值。还是按照中序遍历的顺序遍历二叉搜索树，每遍历到一个节点就在链表中插入与树中节点的

然后将指针P指向链表的头节点，接下来函数hasNext和next的执行过程与前一种方法一样。在完成迭代器的初始化之后，链表中将包含n个节点。这个链表在初始化完成之后一直存在

• 缺点：空间复杂度较高，为O（n），因为需要将所有节点或元素预先保存到list中

原地遍历

可以在不修改输入的二叉搜索树的同时优化空间效率

在非递归遍历的过程中：

• hasNext： 根据stack是否不为空，判断是否有下一个元素

• next： 用正常的方式取出下一个元素

这种做法能做到既不修改原有树，也让空间复杂度降为 O(h) ，即数的高度量级

代码

class BSTIterator {
    static class Command{
        public String commond;
        public TreeNode treeNode;
        public Command(String com,TreeNode node){
            this.commond = com;
            this.treeNode = node;
        }
    }
        
    private Stack<Command> stack;

    public BSTIterator(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        stack = new Stack<>();
        stack.push(new Command("go",root));
    }

    public int next() {
        while (!stack.isEmpty()) {
            Command pop = stack.pop();
            if ("go".equals(pop.commond)) {
                if (pop.treeNode.right != null) {
                    stack.push(new Command("go", pop.treeNode.right));
                }
                stack.push(new Command("collect", pop.treeNode));
                if (pop.treeNode.left != null) {
                    stack.push(new Command("go", pop.treeNode.left));
                }

            } else {
                return pop.treeNode.val;
            }
        }
        return 0;
    }

    public boolean hasNext() {
        return !stack.isEmpty();
    }
}