Binary Search Tree Iterator 理解

Binary Search Tree Iterator 有一行代码一直不理解，为什么要声明        
        TreeNode curt = stack.peek();
        TreeNode node = curt;
直接 声明  TreeNode node = stack.peek(); 然后返回node.val不行吗？ 谢谢大家

public class BSTIterator {
    private Stack<TreeNode> stack = new Stack<>();

    // @param root: The root of binary tree.
    public BSTIterator(TreeNode root) {
        while (root != null) {
            stack.push(root);
            root = root.left;
        }
    }

    //@return: True if there has next node, or false
    public boolean hasNext() {
        return !stack.isEmpty();
    }

    //@return: return next node
    public TreeNode next() {
        TreeNode curt = stack.peek();
        TreeNode node = curt;

        // move to the next node
        if (node.right == null) {
            node = stack.pop();
            while (!stack.isEmpty() && stack.peek().right == node) {
                node = stack.pop();
            }
        } else {
            node = node.right;
            while (node != null) {
                stack.push(node);
                node = node.left;
            }
        }

        return curt;
    }
}

node在下面一个if else之后就不一定还是curt了啊。

gavin92 发表于 2018-12-30 09:31
node在下面一个if else之后就不一定还是curt了啊。

谢谢回复，没太明白，可不可以详细说下，非常感谢
public class BSTIterator {
    private Stack<TreeNode> stack = new Stack<>();

    // @param root: The root of binary tree.
    public BSTIterator(TreeNode root) {
        while (root != null) {
            stack.push(root);
            root = root.left;
        }
    }

    //@return: True if there has next node, or false
    public boolean hasNext() {
        return !stack.isEmpty();
    }

    //@return: return next node
    public TreeNode next() {
        TreeNode node= stack.peek();

        // move to the next node
        if (node.right == null) {
            node = stack.pop();
            while (!stack.isEmpty() && stack.peek().right == node) {
                node = stack.pop();
            }
        } else {
            node = node.right;
            while (node != null) {
                stack.push(node);
                node = node.left;
            }
        }

        return node;
    }
}

limao123 发表于 2018/12/30 11:31:28


谢谢回复，没太明白，可不可以详细说下，非常感谢
public class BSTIterator {
    private Stack<TreeNode> stack = new Stack...

我建议你先建一个简单的树跑跑看，尝试看看结果是不是对的。
我怀疑你不太明白这个iterator是怎么work的，你可以开个IDE，单步调试一步步看看几个变量都是怎么变的。

你把返回的node改掉了啊。。
while (!stack.isEmpty() && stack.peek().right == node) {
                node = stack.pop();
            }
这是在作甚。。

Airtnp 发表于 2018-12-30 13:42
你把返回的node改掉了啊。。
while (!stack.isEmpty() && stack.peek().right == node) {
               ...

想明白了 非常感谢

gavin92 发表于 2018-12-30 12:29
我建议你先建一个简单的树跑跑看，尝试看看结果是不是对的。
我怀疑你不太明白这个iterator是怎么work的 ...

试了一下，我以前理解错了，非常感谢