leetcode之二叉搜索树迭代器（C++）

参考链接

1. https://leetcode-cn.com/problems/binary-search-tree-iterator/
2. https://leetcode-cn.com/problems/binary-search-tree-iterator/solution/er-cha-sou-suo-shu-die-dai-qi-by-leetcod-4y0y/

题目描述

实现一个二叉搜索树迭代器类BSTIterator ，表示一个按中序遍历二叉搜索树（BST）的迭代器：
BSTIterator(TreeNode root) 初始化 BSTIterator 类的一个对象。BST 的根节点 root 会作为构造函数的一部分给出。指针应初始化为一个不存在于 BST 中的数字，且该数字小于 BST 中的任何元素。
boolean hasNext() 如果向指针右侧遍历存在数字，则返回 true ；否则返回 false 。
int next()将指针向右移动，然后返回指针处的数字。
注意，指针初始化为一个不存在于 BST 中的数字，所以对 next() 的首次调用将返回 BST 中的最小元素。

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

解题思路

数组

最朴素的解法，将二叉搜索树进行中序遍历，并将值存入数组。之后的访问操作针对数组即可。

栈

不用事先把所有节点都遍历完，可以用栈一般遍历一边访问，节省空间。

代码

数组

class BSTIterator {
private:
    void traverse(TreeNode* root, vector<int>& res)
    {
        if (root == nullptr)
        {
            return;
        }
        traverse(root->left, res);
        res.push_back(root->val);
        traverse(root->right, res);
    }

    vector<int> inorder(TreeNode* root)
    {
        vector<int> res;
        traverse(root, res);
        return res;
    }

    vector<int> arr;
    int idx;
public:
    BSTIterator(TreeNode* root): idx(0), arr(inorder(root)) {

    }
    
    int next() {
        return arr[idx++];
    }
    
    bool hasNext() {
        return idx < arr.size();
    }
};

栈

class BSTIterator {
private:
    TreeNode* cur;
    stack<TreeNode*> stk;
public:
    BSTIterator(TreeNode* root): cur(root) {}
    
    int next() {
        while (cur != nullptr) {
            stk.push(cur);
            cur = cur->left;
        }
        cur = stk.top();
        stk.pop();
        int ret = cur->val;
        cur = cur->right;
        return ret;
    }
    
    bool hasNext() {
        return cur != nullptr || !stk.empty();
    }
};