文章目錄

• • 1. 題目信息
  • 2. 二叉樹中序遍歷

1. 題目信息

實現一個二叉搜索樹迭代器。你將使用二叉搜索樹的根節點初始化迭代器。

調用 next() 將返回二叉搜索樹中的下一個最小的數。

示例：

BSTIterator iterator = new BSTIterator(root); iterator.next(); // 返回 3 iterator.next(); // 返回 7 iterator.hasNext(); // 返回 true iterator.next(); // 返回 9 iterator.hasNext(); // 返回 true iterator.next(); // 返回 15 iterator.hasNext(); // 返回 true iterator.next(); // 返回 20 iterator.hasNext(); // 返回 false

提示：

next() 和 hasNext() 操作的時間復雜度是 O(1)，并使用 O(h) 內存，其中 h 是樹的高度。
你可以假設 next() 調用總是有效的，也就是說，當調用 next() 時，BST 中至少存在一個下一個最小的數。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/binary-search-tree-iterator
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 二叉樹中序遍歷

• 搜索樹中序是非降的
  

class BSTIterator {stack<TreeNode*> stk;TreeNode *cur; public:BSTIterator(TreeNode* root) {cur = root;}/** @return the next smallest number */int next() {if(!hasNext())return -1;while(cur){stk.push(cur);cur = cur->left;}cur = stk.top();int v = cur->val;stk.pop();cur = cur->right;return v;}/** @return whether we have a next smallest number */bool hasNext() {return !stk.empty() || cur;} };/*** Your BSTIterator object will be instantiated and called as such:* BSTIterator* obj = new BSTIterator(root);* int param_1 = obj->next();* bool param_2 = obj->hasNext();*/

總結

以上是生活随笔為你收集整理的LeetCode 173. 二叉搜索树迭代器（中序遍历）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。