设计

根据二叉树的中序遍历的迭代解法，稍改代码，就是本题的解法。

初始化 : 传入了根结点，根据迭代思路，将结点的左链依次入栈。

nextnextnext : 栈顶结点就是所求。根据迭代思路，当前结点要变成栈顶结点的右儿子。由于 nextnextnext 只想取个值，我们要帮他找到下一个中序结点，将当前结点的左链依次入栈。

hasnexthasnexthasnext : 结点按中序遍历入栈，看栈内有没有元素，就知道有没有取完结点了。

class BSTIterator {
public:stack stk;BSTIterator(TreeNode* root) {while(root){stk.push(root);root = root->left;}}int next(){auto t = stk.top();stk.pop();int ans = t -> val;t = t->right;while(t){stk.push(t);t = t->left;}return ans;}bool hasNext() {return stk.size();}
};

1. 时间复杂度 : O(1)O(1)O(1) ， hhh 是二叉树的最大深度， 对于每个查询， nextnextnext 最坏时间复杂度 O(h)O(h)O(h) ，对应走 hhh 的深度才找到下一个结点。如果查询整个二叉树，每个结点最多入栈一次， nextnextnext 均摊时间复杂度 O(1)O(1)O(1) 。
2. 空间复杂度 : O(h)O(h)O(h) ， 最坏空间复杂度 O(h)O(h)O(h) 。

AC

致语

• 理解思路很重要！
• 欢迎读者在评论区留言，墨染看到就会回复的。