每日一题之leetcode173. 二叉搜索树迭代器

题目描述：

实现一个二叉搜索树迭代器类BSTIterator ，表示一个按中序遍历二叉搜索树（BST）的迭代器：

• BSTIterator(TreeNode root) 初始化 BSTIterator 类的一个对象。BST 的根节点 root 会作为构造函数的一部分给出。指针应初始化为一个不存在于 BST 中的数字，且该数字小于 BST 中的任何元素。
• boolean hasNext() 如果向指针右侧遍历存在数字，则返回 true ；否则返回 false 。
• int next()将指针向右移动，然后返回指针处的数字。

注意，指针初始化为一个不存在于 BST 中的数字，所以对 next() 的首次调用将返回 BST 中的最小元素。

你可以假设 next() 调用总是有效的，也就是说，当调用 next() 时，BST 的中序遍历中至少存在一个下一个数字。

题目示例：

 c++代码：

#include <stack>

#include <limits>

using namespace std;

struct TreeNode {

    int Val;

    TreeNode *Left;

    TreeNode *Right;

}

class BSTIterator {

    private:

    stack<TreeNode*> nodes;  //存储节点的栈

    void pushAllLeft(TreeNode* root) {

        //将当前节点及其左子节点依次入栈

        while(root != nullptr) {

            nodes.push(root);

            root = root->Left;

        }

    }

    public:

    BSTIterator(TreeNode* root) {

        //在构造函数中将根节点及其左子节点入栈

        pushAllLeft(root);

    }

    int next() {

        TreeNode* root = nodes.top();

        nodes.pop();//弹出栈顶节点

        if (root->Right != nullptr) {

            pushAllLeft(root->Right); //将右子树的左子节点依次入栈

        }

        return root->Val;//返回当前节点的值

    }

    bool haxNext() {

        return !nodes.empty(); //检查栈是否为空，判断是否还有下一个元素

    }

}

go代码：

type TreeNode struct {
    Val   int
    Left  *TreeNode
    Right *TreeNode
}

type BSTIterator struct {
    nodes []*TreeNode // 存储节点的切片
}

func Constructor(root *TreeNode) BSTIterator {
    it := BSTIterator{}
    it.pushAllLeft(root) // 将根节点及其左子节点入栈
    return it
}

func (it *BSTIterator) pushAllLeft(node *TreeNode) {
    // 将当前节点及其左子节点依次入栈
    for node != nil {
        it.nodes = append(it.nodes, node)
        node = node.Left
    }
}

func (it *BSTIterator) Next() int {
    node := it.nodes[len(it.nodes)-1] // 取出切片末尾的节点
    it.nodes = it.nodes[:len(it.nodes)-1] // 切片缩小，弹出末尾的节点
    if node.Right != nil {
        it.pushAllLeft(node.Right) // 将右子树的左子节点依次入栈
    }
    return node.Val // 返回当前节点的值
}

func (it *BSTIterator) HasNext() bool {
    return len(it.nodes) > 0 // 检查切片是否为空，判断是否还有下一个元素
}